Move pipeline to CUDA in ZeroGPU function

app.py

@@ -52,7 +52,7 @@ def generate(
     )
 
     logger.debug(f"Loading pipeline: {dict(model=model)}")
-    pipe = pipeline_type.from_pretrained(model).to("cuda")
+    pipe = pipeline_type.from_pretrained(model)
 
     logger.debug(f"Generating image: {dict(prompt=prompt)}")
     additional_args = (
@@ -62,7 +62,7 @@ def generate(
     gpu_runner = gpu_3min if model == "timbrooks/instruct-pix2pix" else gpu
 
     images = gpu_runner(
-        lambda: pipe(
+        lambda: pipe.to("cuda")(
             prompt=prompt,
             image=init_image,
             callback_on_step_end=progress_callback,