moving pipe_inversion to cpu after first inversion to save memory

src/editor.py

@@ -47,22 +47,26 @@ class ImageEditorDemo:
         noise = [randn_tensor(latents_size, dtype=torch.float16, device=torch.device(device), generator=g_cpu) for i
                  in range(cfg.num_inversion_steps)]
         pipe_inversion.scheduler.set_noise_list(noise)
-        pipe_inference.scheduler.set_noise_list(noise)
         pipe_inversion.scheduler_inference.set_noise_list(noise)
         pipe_inversion.set_progress_bar_config(disable=True)
-        pipe_inference.set_progress_bar_config(disable=True)
         self.cfg = cfg
         self.pipe_inversion.cfg = cfg
         self.pipe_inference.cfg = cfg
         self.inv_hp = inv_hp  # [2, 0.1, 0.2]
         self.edit_cfg = cfg.edit_guidance_scale
 
-        self.pipe_inference.to(device)
         self.pipe_inversion.to(device)
-
         self.last_latent = self.invert(self.original_image, description_prompt)
         self.original_latent = self.last_latent
 
+        if device  == 'cuda':
+            # after the inversion, we can move the inversion model to the CPU
+            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)
+
     def invert(self, init_image, base_prompt):
         res = self.pipe_inversion(prompt=base_prompt,
                              num_inversion_steps=self.cfg.num_inversion_steps,