tianleliphoebe commited on
Commit
817acc8
1 Parent(s): 8a702cc

Update model/model_manager.py

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -78,7 +78,7 @@ class ModelManager:
78
  results = [future.result() for future in futures]
79
  return results[0], results[1], model_names[0], model_names[1]
80
 
81
- @spaces.GPU(duration=200)
82
  def generate_video_vg(self, prompt, model_name):
83
  pipe = self.load_model_pipe(model_name)
84
  result = pipe(prompt=prompt)
 
78
  results = [future.result() for future in futures]
79
  return results[0], results[1], model_names[0], model_names[1]
80
 
81
+ @spaces.GPU(duration=120)
82
  def generate_video_vg(self, prompt, model_name):
83
  pipe = self.load_model_pipe(model_name)
84
  result = pipe(prompt=prompt)