Spaces:
Runtime error
Runtime error
Update src/editor.py
Browse files- 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)
|