barakmeiri commited on
Commit
5666f2a
·
verified ·
1 Parent(s): 424f073

Update src/editor.py

Browse files

update noise list device

Files changed (1) hide show
  1. src/editor.py +4 -0
src/editor.py CHANGED
@@ -91,4 +91,8 @@ class ImageEditorDemo:
91
  self.pipe_inversion.to(device)
92
  self.last_latent.to(device)
93
  self.original_latent.to(device)
 
 
 
 
94
  return self
 
91
  self.pipe_inversion.to(device)
92
  self.last_latent.to(device)
93
  self.original_latent.to(device)
94
+
95
+ self.pipe_inversion.scheduler.set_noise_list_device(device)
96
+ self.pipe_inference.scheduler.set_noise_list_device(device)
97
+ self.pipe_inversion.scheduler_inference.set_noise_list_device(device)
98
  return self