kiwhansong commited on
Commit
486c60b
·
1 Parent(s): 5359939

fix navigation

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -163,6 +163,7 @@ class CustomProgressBar:
163
  return getattr(self.pbar, attr)
164
 
165
 
 
166
  @torch.autocast("cuda")
167
  @torch.no_grad()
168
  def navigate_video(
@@ -570,7 +571,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="teal")) as demo:
570
  gr.Markdown(
571
  f"""
572
  ## Demo 2: Single Image → Long {LONG_LENGTH}-second Video
573
- > #### _Diffusion Forcing Transformer, with History Guidance, generates long videos via sliding window rollouts and temporal super-resolution._
574
  """
575
  )
576
 
 
163
  return getattr(self.pbar, attr)
164
 
165
 
166
+ @spaces.GPU()
167
  @torch.autocast("cuda")
168
  @torch.no_grad()
169
  def navigate_video(
 
571
  gr.Markdown(
572
  f"""
573
  ## Demo 2: Single Image → Long {LONG_LENGTH}-second Video
574
+ > #### _Diffusion Forcing Transformer, with History Guidance, can generate long videos via sliding window rollouts and temporal super-resolution._
575
  """
576
  )
577