fix alignment-handbook version

This commit is contained in:
Yu Meng
2024-08-04 23:12:21 -04:00
parent 481df8ed57
commit 1aa921d748
8 changed files with 903 additions and 2 deletions
+1 -1
View File
@@ -218,7 +218,7 @@ def main():
use_cache=False if training_args.gradient_checkpointing else True,
device_map=get_kbit_device_map() if quantization_config is not None else None,
quantization_config=quantization_config,
attn_implementation=training_args.attn_implementation,
attn_implementation=model_args.attn_implementation,
)
model = model_args.model_name_or_path
-1
View File
@@ -69,4 +69,3 @@ class SimPOConfig(TrainingArguments):
dataset_num_proc: Optional[int] = None
attn_implementation: str = None