423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
Jonery
2024-06-18 12:39:26 +08:00
parent 0f72aac8c9
commit 8f7c78b641
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/hparams/parser.py
View File

@@ -219,7 +219,7 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:
and training_args.deepspeed_plugin is not None and training_args.deepspeed_plugin is not None
and training_args.deepspeed_plugin.zero_stage < 3 and training_args.deepspeed_plugin.zero_stage < 3
): ):
raise ValueError(f"Layer-wise BAdam only supports DeepSpeed ZeRO 3 stage, got stage {self.args.deepspeed_plugin.zero_stage}") raise ValueError(f"Layer-wise BAdam only supports DeepSpeed ZeRO 3 stage, got stage {training_args.deepspeed_plugin.zero_stage}")
if (finetuning_args.use_galore) and training_args.deepspeed is not None: if (finetuning_args.use_galore) and training_args.deepspeed is not None:
raise ValueError("GaLore are incompatible with DeepSpeed yet.") raise ValueError("GaLore are incompatible with DeepSpeed yet.")