mirror of
https://github.com/FlagOpen/FlagEmbedding.git
synced 2025-07-24 09:25:28 +00:00
15 lines
404 B
Python
15 lines
404 B
Python
|
from dataclasses import dataclass, field
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ModelArguments:
|
||
|
|
model_name_or_path: str = field(
|
||
|
|
default='Shitao/flag-text-embedding-chinese', 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"}
|
||
|
|
)
|