yuanshengni commited on
Commit
eb421e6
1 Parent(s): fff5214

Update model/model_manager.py

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -71,7 +71,7 @@ class ModelManager:
71
  results = [future.result() for future in futures]
72
  return results[0], results[1], model_names[0], model_names[1]
73
 
74
- @spaces.GPU(duration=180)
75
  def generate_video_vg(self, prompt, model_name):
76
  pipe = self.load_model_pipe(model_name)
77
  result = pipe(prompt=prompt)
 
71
  results = [future.result() for future in futures]
72
  return results[0], results[1], model_names[0], model_names[1]
73
 
74
+ @spaces.GPU(duration=200)
75
  def generate_video_vg(self, prompt, model_name):
76
  pipe = self.load_model_pipe(model_name)
77
  result = pipe(prompt=prompt)