wassname/SimPO
1
0
Fork 0
mirror of https://github.com/wassname/SimPO.git synced 2026-09-12 12:11:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

update trainer argument

Browse Source
This commit is contained in:
Yu Meng
2024-07-13 16:50:42 -04:00
parent 6c3757f3f2
commit 3e9c4cc3bd
8 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/simpo_config.py
+2
View File
@@ -68,3 +68,5 @@ class SimPOConfig(TrainingArguments):
model_init_kwargs: Optional[Dict] = None
dataset_num_proc: Optional[int] = None
attn_implementation: str = None