fix sdxl_smooth_quant

xin3he · xin3he · commit d2435815c0e4 · 2025-02-22T15:35:11.000+08:00
Signed-off-by: xin3he &lt;xin3.he@intel.com&gt;
diff --git a/examples/3.x_api/pytorch/diffusion_model/diffusers/stable_diffusion/smooth_quant/sdxl_smooth_quant.py b/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

Original file line number	Diff line number	Diff line change
`@@ -361,6 +361,7 @@ def main():`
`361`	`361`	`torch_dtype=dtype,`
`362`	`362`	`use_safetensors=True,`
`363`	`363`	`)`
	`364`	`+ pipeline = pipeline.to(dtype) # Ensure all modules are set as dtype`
`364`	`365`	`pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)`
`365`	`366`
`366`	`367`	`# This is a list of prompts`