aducsdr commited on
Commit
7c96174
·
verified ·
1 Parent(s): ae8872e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()