Skip to content

GitLab

Commit f06e4e55 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 57f7d259
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py
View file @ f06e4e55
...@@ -214,7 +214,7 @@ class AltDiffusionPipeline(DiffusionPipeline): ...@@ -214,7 +214,7 @@ class AltDiffusionPipeline(DiffusionPipeline):
cpu_offload(cpu_offloaded_model, device) cpu_offload(cpu_offloaded_model, device)
if self.safety_checker is not None: 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 @property
def _execution_device(self): def _execution_device(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment