Former-commit-id: c9fdef10de737d1f433209812ef73e29cb60490a
This commit is contained in:
hiyouga 2024-06-28 00:41:58 +08:00
parent 95bf795de4
commit 9103fdf866

View File

@ -202,9 +202,6 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:
if is_deepspeed_zero3_enabled():
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 (
finetuning_args.use_galore
and finetuning_args.galore_layerwise