Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ def generation_step(runner, text_embeds_dict, cond_latents):
|
|
141 |
video_tensors = runner.inference(noises=noises, conditions=conditions, **text_embeds_dict)
|
142 |
return [rearrange(v, "c t h w -> t c h w") for v in video_tensors]
|
143 |
|
144 |
-
|
145 |
def generation_loop(video_path, seed=666, fps_out=24):
|
146 |
if video_path is None: return None, None, None
|
147 |
runner = configure_runner()
|
|
|
141 |
video_tensors = runner.inference(noises=noises, conditions=conditions, **text_embeds_dict)
|
142 |
return [rearrange(v, "c t h w -> t c h w") for v in video_tensors]
|
143 |
|
144 |
+
@spaces.GPU
|
145 |
def generation_loop(video_path, seed=666, fps_out=24):
|
146 |
if video_path is None: return None, None, None
|
147 |
runner = configure_runner()
|