mirror of
https://github.com/wassname/SimPO.git
synced 2026-07-16 11:16:27 +08:00
fix alignment-handbook version
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -69,4 +69,3 @@ class SimPOConfig(TrainingArguments):
|
||||
|
||||
dataset_num_proc: Optional[int] = None
|
||||
|
||||
attn_implementation: str = None
|
||||
|
||||
Reference in New Issue
Block a user