Skip to content

Commit 30b1ef2

Browse files
committed
condition better.
1 parent 1d46b21 commit 30b1ef2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/diffusers/pipelines/pipeline_loading_utils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -764,7 +764,11 @@ def load_sub_model(
764764
else:
765765
loading_kwargs["low_cpu_mem_usage"] = False
766766

767-
if quantization_config is not None and isinstance(quantization_config, PipelineQuantizationConfig):
767+
if (
768+
quantization_config is not None
769+
and isinstance(quantization_config, PipelineQuantizationConfig)
770+
and issubclass(class_obj, torch.nn.Module)
771+
):
768772
exclude_modules = quantization_config.exclude_modules or []
769773
if name not in exclude_modules:
770774
model_quant_config = _resolve_quant_config(quantization_config, is_diffusers=is_diffusers_model)

0 commit comments

Comments
 (0)