Skip to content

Commit f06e4e5

Browse files
make style
1 parent 57f7d25 commit f06e4e5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ def enable_sequential_cpu_offload(self, gpu_id=0):
214214
cpu_offload(cpu_offloaded_model, device)
215215

216216
if self.safety_checker is not None:
217-
cpu_offload(self.safety_checker, device, offload_buffers=True)
217+
cpu_offload(self.safety_checker, execution_device=device, offload_buffers=True)
218218

219219
@property
220220
def _execution_device(self):

0 commit comments

Comments
 (0)