move pipes explicitly to 'cpu' after using them

app.py

@@ -120,6 +120,7 @@ with gr.Blocks(css="style.css") as demo:
         inversion_state = ImageEditorDemo.invert(pipe_inversion.to(device), input_image, description_prompt, config,
                                                  [rnri_iterations, rnri_alpha, rnri_lr], device)
         if device == 'cuda':
+            pipe_inversion.to('cpu')
             torch.cuda.empty_cache()
             # pipe_inversion = pipe_inversion.to('cpu')
         print(f"#### 3 #### pipe_inversion.device: {pipe_inversion.device}")
@@ -158,6 +159,7 @@ with gr.Blocks(css="style.css") as demo:
                                      inversion_state['cfg'], inversion_state['cfg'].edit_guidance_scale)
 
         if device == 'cuda':
+            pipe_inference.to('cpu')
             torch.cuda.empty_cache()