1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

handle offload_state_dict when initing transformers models (#12438)

This commit is contained in:
Sayak Paul
2025-10-07 13:51:20 +05:30
committed by sayakpaul
parent 0f252be0ed
commit 08782bf3bf

View File

@@ -830,6 +830,9 @@ def load_sub_model(
else:
loading_kwargs["low_cpu_mem_usage"] = False
if is_transformers_model and is_transformers_version(">=", "4.57.0"):
loading_kwargs.pop("offload_state_dict")
if (
quantization_config is not None
and isinstance(quantization_config, PipelineQuantizationConfig)