cuda

app.py

@@ -25,12 +25,12 @@ pipe = VSFStableDiffusion3Pipeline.from_pretrained(
     "stabilityai/stable-diffusion-3.5-large-turbo",
     torch_dtype=torch.bfloat16,
     hf_token=os.environ.get("HF_TOKEN", None)
-).to("cuda")
+)
 import os
 @spaces.GPU
 def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step, seed, progress=gr.Progress(track_tqdm=False)):
     lambda total: progress.tqdm(range(total))
-        
+    pipe = pipe.to("cuda")
     print(f"Generating image with params: {positive_prompt}, {negative_prompt}, {guidance_scale}, {bias}, {step}")
 
     output = pipe(