This commit is contained in:
hiyouga 2024-06-28 00:41:58 +08:00
parent 0f421055da
commit 8ed6b367e2
1 changed files with 0 additions and 3 deletions

View File

@ -202,9 +202,6 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:
if is_deepspeed_zero3_enabled(): if is_deepspeed_zero3_enabled():
raise ValueError("`pure_bf16` is incompatible with DeepSpeed ZeRO-3.") raise ValueError("`pure_bf16` is incompatible with DeepSpeed ZeRO-3.")
if training_args.fp16 or training_args.bf16:
raise ValueError("Turn off mixed precision training when using `pure_bf16`.")
if ( if (
finetuning_args.use_galore finetuning_args.use_galore
and finetuning_args.galore_layerwise and finetuning_args.galore_layerwise