Update app.py

app.py

@@ -2159,8 +2159,8 @@ def update_selection(evt: gr.SelectData, width, height):
 
 @spaces.GPU(duration=100)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
-    pipe.to("cuda")
-    generator = torch.Generator(device="cuda").manual_seed(seed)
+    pipe.to("cpu")
+    generator = torch.Generator(device="cpu").manual_seed(seed)
     with calculateDuration("Generating image"):
         # Generate image
         for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
@@ -2177,8 +2177,8 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
             yield img
 
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, lora_scale, seed):
-    generator = torch.Generator(device="cuda").manual_seed(seed)
-    pipe_i2i.to("cuda")
+    generator = torch.Generator(device="cpu").manual_seed(seed)
+    pipe_i2i.to("cpu")
     image_input = load_image(image_input_path)
     final_image = pipe_i2i(
         prompt=prompt_mash,