return not inplace .to(device)

src/editor.py

@@ -55,17 +55,17 @@ class ImageEditorDemo:
         self.inv_hp = inv_hp  # [2, 0.1, 0.2]
         self.edit_cfg = cfg.edit_guidance_scale
 
-        self.pipe_inversion.to(device)
+        self.pipe_inversion = self.pipe_inversion.to(device)
         self.last_latent = self.invert(self.original_image, description_prompt)
         self.original_latent = self.last_latent
 
-        if device  == 'cuda':
+        # if device  == 'cuda':
             # after the inversion, we can move the inversion model to the CPU
-            self.pipe_inversion.to('cpu')
+        self.pipe_inversion = self.pipe_inversion.to('cpu')
 
         pipe_inference.scheduler.set_noise_list(noise)
         pipe_inference.set_progress_bar_config(disable=True)
-        self.pipe_inference.to(device)
+        self.pipe_inference = self.pipe_inference.to(device)
 
     def invert(self, init_image, base_prompt):
         res = self.pipe_inversion(prompt=base_prompt,