fix sdxl_smooth_quant

Signed-off-by: xin3he <xin3.he@intel.com>

Author: xin3he

examples/3.x_api/pytorch/diffusion_model/diffusers/stable_diffusion/smooth_quant/main.py

@@ -400,6 +400,7 @@ def __call__(
     torch_dtype=dtype,
     use_safetensors=True,
 )
+pipe = pipe.to(dtype)  # Ensure all modules are set as dtype
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 
 if args.refiner:

examples/3.x_api/pytorch/diffusion_model/diffusers/stable_diffusion/smooth_quant/sdxl_smooth_quant.py

@@ -361,6 +361,7 @@ def main():
         torch_dtype=dtype,
         use_safetensors=True,
     )
+    pipeline = pipeline.to(dtype)  # Ensure all modules are set as dtype
     pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
 
     # This is a list of prompts

neural_compressor/torch/quantization/config.py

@@ -2096,7 +2096,7 @@ def __new__(
         else:
             config_cls = self._model_mapping[STATIC_QUANT]
         return config_cls(*args, **kwargs)
-    
+
     @classmethod
     def get_config_set_for_tuning(cls, dtype="int8"):
         """Map to different config set for tuning."""