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 1d46b21 commit 30b1ef2Copy full SHA for 30b1ef2
src/diffusers/pipelines/pipeline_loading_utils.py
@@ -764,7 +764,11 @@ def load_sub_model(
764
else:
765
loading_kwargs["low_cpu_mem_usage"] = False
766
767
- if quantization_config is not None and isinstance(quantization_config, PipelineQuantizationConfig):
+ if (
768
+ quantization_config is not None
769
+ and isinstance(quantization_config, PipelineQuantizationConfig)
770
+ and issubclass(class_obj, torch.nn.Module)
771
+ ):
772
exclude_modules = quantization_config.exclude_modules or []
773
if name not in exclude_modules:
774
model_quant_config = _resolve_quant_config(quantization_config, is_diffusers=is_diffusers_model)
0 commit comments