diff --git a/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py b/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py index e9f61b4e48..06919596df 100644 --- a/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py +++ b/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py @@ -214,7 +214,7 @@ class AltDiffusionPipeline(DiffusionPipeline): cpu_offload(cpu_offloaded_model, device) if self.safety_checker is not None: - cpu_offload(self.safety_checker, device, offload_buffers=True) + cpu_offload(self.safety_checker, execution_device=device, offload_buffers=True) @property def _execution_device(self):