Update src/editor.py

add to(device) to ImageEditor

src/editor.py

@@ -86,3 +86,8 @@ class ImageEditorDemo:
                             guidance_scale=self.edit_cfg).images[0]
         return image
 
+    def to(device):
+        self.pipe_inference.to(device)
+        self.pipe_inversion.to(device)
+        self.last_latent.to(device)
+        self.original_latent.to(device)