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 GitHub
parent 0974b4c606
commit 2d69bacb00

View File

@@ -838,6 +838,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)