when sitting pipe, move to cpu and then to device

app.py

@@ -60,8 +60,8 @@ with gr.Blocks(css="style.css") as demo:
     @spaces.GPU
     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')
+        if image_editor is not None:
+            image_editor = image_editor.to('cpu')
 
         if isinstance(num_inference_steps, str): num_inference_steps = int(num_inference_steps) 
         if isinstance(num_inversion_steps, str): num_inversion_steps = int(num_inversion_steps)