INNER CODE UNIT · Python

ModelArguments

HIT-SCIR/Chinese-Mixtral-8x7B · train.py:23

class ModelArguments:
    model_name_or_path: Optional[str] = field(
        default=None,
        metadata={"help": "Path to model"},
    )
    tokenizer_name_or_path: Optional[str] = field(
        default=None,
        metadata={"help": "Path to tokenizer. "
                          "We distinguish this setting from `model_name_or_path` "
                          "so that datasets will not generate redundant cache."},
    )


@dataclass
class DataArguments:
    train_datasets: List[str] = field(
        default=None,
        metadata={"help": "Training dataset ratio and names. "

View source record →