minor bug fix

app.py

@@ -110,15 +110,15 @@ with gr.Blocks(css="style.css") as demo:
                            inversion_max_step=inversion_max_step)
         if device == 'cuda':
             torch.cuda.empty_cache()
-            pipe_inversion = pipe_inversion.to(device)
+            # pipe_inversion = pipe_inversion.to(device)
             # if image_editor is not None:
             #     image_editor = image_editor.to('cpu')
 
-        inversion_state = ImageEditorDemo.invert(pipe_inversion, input_image, description_prompt, config,
+        inversion_state = ImageEditorDemo.invert(pipe_inversion.to(device), input_image, description_prompt, config,
                                                  [rnri_iterations, rnri_alpha, rnri_lr], device)
         if device == 'cuda':
             torch.cuda.empty_cache()
-            pipe_inversion = pipe_inversion.to('cpu')
+            # pipe_inversion = pipe_inversion.to('cpu')
         return inversion_state, "Input has set!"
 
 
@@ -144,12 +144,10 @@ with gr.Blocks(css="style.css") as demo:
 
         if device == 'cuda':
             torch.cuda.empty_cache()
-            pipe_inference = pipe_inference.to(device)
-        image = ImageEditorDemo.edit(pipe_inference, target_prompt, inversion_state['latent'], inversion_state['noise'],
+        image = ImageEditorDemo.edit(pipe_inference.to(device), target_prompt, inversion_state['latent'], inversion_state['noise'],
                                      inversion_state['cfg'], inversion_state['cfg'].edit_guidance_scale)
 
         if device == 'cuda':
-            pipe_inference = pipe_inference.to('cpu')
             torch.cuda.empty_cache()
         return image