condition better.

huggingface · Mar 10, 2025 · 30b1ef2 · 30b1ef2
1 parent 1d46b21
commit 30b1ef2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/diffusers/pipelines/pipeline_loading_utils.py b/src/diffusers/pipelines/pipeline_loading_utils.py
@@ -764,7 +764,11 @@ def load_sub_model(
         else:
             loading_kwargs["low_cpu_mem_usage"] = False
 
-    if quantization_config is not None and isinstance(quantization_config, PipelineQuantizationConfig):
+    if (
+        quantization_config is not None
+        and isinstance(quantization_config, PipelineQuantizationConfig)
+        and issubclass(class_obj, torch.nn.Module)
+    ):
         exclude_modules = quantization_config.exclude_modules or []
         if name not in exclude_modules:
             model_quant_config = _resolve_quant_config(quantization_config, is_diffusers=is_diffusers_model)