alex commited on
Commit
12d3925
·
1 Parent(s): e5f986b

duration fixed

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -157,7 +157,7 @@ def get_duration(
157
  return int(sample_steps * 3 + warmup)
158
 
159
 
160
- @spaces.GPU(duration=160)
161
  def generate_video(
162
  text_prompt,
163
  image,
 
157
  return int(sample_steps * 3 + warmup)
158
 
159
 
160
+ @spaces.GPU(duration=get_duration)
161
  def generate_video(
162
  text_prompt,
163
  image,