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()
|