| 35 | |
| 36 | @dataclass |
| 37 | class TrainingArguments(transformers.Seq2SeqTrainingArguments): |
| 38 | cache_dir: Optional[str] = field(default=None) |
| 39 | optim: str = field(default="adamw_torch") |
| 40 | lora_enable: bool = False |
| 41 | lora_rank: int = 64 |
| 42 | lora_alpha: int = 32 |
| 43 | lora_dropout: float = 0.05 |
| 44 | |
| 45 | |
| 46 | @dataclass |
nothing calls this directly
no outgoing calls
no test coverage detected