make style

f06e4e55 · Patrick von Platen · 57f7d259 · f06e4e55
Commit f06e4e55 authored Jan 12, 2023 by Patrick von Platen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py ...ffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py +1 -1

No files found.
--- 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):