AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

make style

Browse Source
This commit is contained in:
Patrick von Platen
2023-01-12 11:33:11 +01:00
parent 57f7d25934
commit f06e4e5579
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py
View File

@@ -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):