Spaces:
Runtime error
Runtime error
Update src/editor.py
Browse filesupdate noise list device
- 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
|