FlagEmbedding/examples/search_demo/arguments.py

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"}
    )