pyramid-flow

Running on Zero

yasserrmd commited on 6 days ago

Commit

a4b6a3e

•

1 Parent(s): b664a31

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ model = load_model()
 # Text-to-video generation function
 @spaces.GPU(duration=120)
-def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
     multiplier = 3
     temp = int(duration * multiplier) + 1  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32

 # Text-to-video generation function
 @spaces.GPU(duration=120)
+def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
     multiplier = 3
     temp = int(duration * multiplier) + 1  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32