From f06e4e5579b2c43a35a1678dea9b9ec3d66d365a Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 12 Jan 2023 11:33:11 +0100 Subject: [PATCH] make style --- src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):