Fix ZeroGPU crash by deferring .to(cuda)

app.py

@@ -10,19 +10,21 @@ pipe_t2i = StableDiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16,
     safety_checker=None
-).to("cuda")
+)
 
 # 2. Instruct-Pix2Pix Model (Instruction-Based Editing)
 pipe_edit = StableDiffusionInstructPix2PixPipeline.from_pretrained(
     "timbrooks/instruct-pix2pix",
     torch_dtype=torch.float16,
     safety_checker=None
-).to("cuda")
+)
 
 print("Models loaded successfully!")
 
 @spaces.GPU
 def generate_or_edit(prompt, input_image, image_guidance_scale, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress(track_tqdm=True)):
+    pipe_t2i.to("cuda")
+    pipe_edit.to("cuda")
     if randomize_seed:
         seed = torch.randint(0, 2**32 - 1, (1,)).item()
     generator = torch.Generator("cuda").manual_seed(int(seed))