Update app.py

app.py

@@ -289,7 +289,6 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             good_vae=good_vae,
         ):
             yield img
-    pipe.to("cpu")
 
 @spaces.GPU(duration=90)
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
@@ -308,7 +307,6 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
         joint_attention_kwargs={"scale": 1.0},
         output_type="pil",
     ).images[0]
-    pipe_i2i.to("cpu")
     return final_image
 
 @spaces.GPU(duration=90)