We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0974b4c commit 2d69bacCopy full SHA for 2d69bac
src/diffusers/pipelines/pipeline_loading_utils.py
@@ -838,6 +838,9 @@ def load_sub_model(
838
else:
839
loading_kwargs["low_cpu_mem_usage"] = False
840
841
+ if is_transformers_model and is_transformers_version(">=", "4.57.0"):
842
+ loading_kwargs.pop("offload_state_dict")
843
+
844
if (
845
quantization_config is not None
846
and isinstance(quantization_config, PipelineQuantizationConfig)
0 commit comments