barakmeiri commited on
Commit
c91e65a
·
verified ·
1 Parent(s): 25a833e

Update src/editor.py

Browse files
Files changed (1) hide show
  1. src/editor.py +1 -1
src/editor.py CHANGED
@@ -86,7 +86,7 @@ class ImageEditorDemo:
86
  guidance_scale=self.edit_cfg).images[0]
87
  return image
88
 
89
- def to(device):
90
  self.pipe_inference.to(device)
91
  self.pipe_inversion.to(device)
92
  self.last_latent.to(device)
 
86
  guidance_scale=self.edit_cfg).images[0]
87
  return image
88
 
89
+ def to(self, device):
90
  self.pipe_inference.to(device)
91
  self.pipe_inversion.to(device)
92
  self.last_latent.to(device)