mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-10-17 00:58:10 +08:00
14 lines
375 B
Python
14 lines
375 B
Python
from typing import Literal, Optional
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass
|
|
class GeneralArguments:
|
|
"""
|
|
Arguments pertaining to which techniques we are going to fine-tuning with.
|
|
"""
|
|
stage: Optional[Literal["pt", "sft", "rm", "ppo"]] = field(
|
|
default="sft",
|
|
metadata={"help": "Which stage will be performed in training."}
|
|
)
|