kiwhansong commited on
Commit
5fa98d9
·
1 Parent(s): 486c60b
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -116,6 +116,7 @@ def single_image_to_long_video(
116
  return export_to_video(gen_video[0].detach().cpu(), fps=fps)
117
 
118
 
 
119
  @torch.autocast("cuda")
120
  @torch.no_grad()
121
  def any_images_to_short_video(
 
116
  return export_to_video(gen_video[0].detach().cpu(), fps=fps)
117
 
118
 
119
+ @spaces.GPU()
120
  @torch.autocast("cuda")
121
  @torch.no_grad()
122
  def any_images_to_short_video(