try fix OOM with HF GPU

app.py

@@ -44,8 +44,10 @@ with gr.Blocks(css="style.css") as demo:
     editor_state = gr.State()
 
     @spaces.GPU
-    def set_pipe(input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
+    def set_pipe(image_editor, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
                  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
+        if image_editor is not None:
+            image_editor = image_editor.to('cpu')
         scheduler_class = MyEulerAncestralDiscreteScheduler
 
         print('\n################## 1')
@@ -172,7 +174,7 @@ with gr.Blocks(css="style.css") as demo:
 
     set_button.click(
         fn=set_pipe,
-        inputs=[input_image, description_prompt, edit_guidance_scale, num_inference_steps,
+        inputs=[editor_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps,
                 num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha, rnri_lr],
         outputs=[editor_state, is_set_text],
     )

src/editor.py

@@ -91,10 +91,10 @@ class ImageEditorDemo:
         return image
 
     def to(self, device):
-        self.pipe_inference.to(device)
-        self.pipe_inversion.to(device)
-        self.last_latent.to(device)
-        self.original_latent.to(device)
+        self.pipe_inference = self.pipe_inference.to(device)
+        self.pipe_inversion = self.pipe_inversion.to(device)
+        self.last_latent = self.last_latent.to(device)
+        self.original_latent = self.original_latent.to(device)
         
         self.pipe_inversion.scheduler.set_noise_list_device(device)
         self.pipe_inference.scheduler.set_noise_list_device(device)