pyramid-flow

Running on Zero

multimodalart HF staff commited on Oct 10

Commit

af7f787

•

1 Parent(s): 9e6bb2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ model = load_model()
 # Text-to-video generation function
 @spaces.GPU(duration=120)
-def generate_video(image, prompt, duration, guidance_scale, video_guidance_scale):
     multiplier = 0.8 if is_canonical else 2.4
     temp = int(duration * 0.8)  # 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(image, prompt, duration, guidance_scale, video_guidance_scale, progress=gr.Progress(track_tqdm=True)):
     multiplier = 0.8 if is_canonical else 2.4
     temp = int(duration * 0.8)  # Convert seconds to temp value (assuming 24 FPS)
     torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32