| 15 | |
| 16 | @dataclass |
| 17 | class ModelArguments: |
| 18 | model_name_or_path: Optional[str] = field(default="facebook/opt-125m") |
| 19 | lora: Optional[bool] = field(default=False) |
| 20 | lora_r: Optional[int] = field(default=16) |
| 21 | lora_alpha: Optional[int] = field(default=16) |
| 22 | lora_dropout: Optional[float] = field(default=0.05) |
| 23 | |
| 24 | |
| 25 | @dataclass |
nothing calls this directly
no outgoing calls
no test coverage detected