mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-06-27 02:39:58 +00:00
17 lines
415 B
Python
17 lines
415 B
Python
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass
|
|
class ModelArguments:
|
|
model_name_or_path: str = field(
|
|
default='BAAI/baai-general-embedding-large-zh',
|
|
metadata={"help": "Path to pretrained model or model identifier from huggingface.co/models"}
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class DataArguments:
|
|
data_path: str = field(
|
|
default='./data', metadata={"help": "Path to wikipedia-22-12"}
|
|
)
|