diff --git a/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py b/src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py index e9f61b4e48cf..06919596df8a 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 @@ def enable_sequential_cpu_offload(self, gpu_id=0): 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):