Skip to content

Commit

Permalink
make style
Browse files Browse the repository at this point in the history
  • Loading branch information
patrickvonplaten committed Jan 12, 2023
1 parent 57f7d25 commit f06e4e5
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
@@ -214,7 +214,7 @@ def enable_sequential_cpu_offload(self, gpu_id=0):
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):

0 comments on commit f06e4e5

Please sign in to comment.