Unverified Commit 29a1c1b4 authored by Kotaro Tanahashi's avatar Kotaro Tanahashi Committed by GitHub
Browse files

Fix `low_cpu_mem_usage` Flag Conflict with DeepSpeed Zero 3 in...

Fix `low_cpu_mem_usage` Flag Conflict with DeepSpeed Zero 3 in `from_pretrained` for Models with `keep_in_fp32_modules`" (#27762)

Fix `from_pretrained` Logic
for `low_cpu_mem_usage` with DeepSpeed Zero3
parent 26ea725b
...@@ -3466,7 +3466,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -3466,7 +3466,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
# Check first if we are `from_pt` # Check first if we are `from_pt`
if use_keep_in_fp32_modules: if use_keep_in_fp32_modules:
if is_accelerate_available(): if is_accelerate_available() and not is_deepspeed_zero3_enabled():
low_cpu_mem_usage = True low_cpu_mem_usage = True
keep_in_fp32_modules = model._keep_in_fp32_modules keep_in_fp32_modules = model._keep_in_fp32_modules
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment