tianleliphoebe commited on
Commit
83e2394
1 Parent(s): 5f52571

Update model/model_manager.py

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -149,7 +149,7 @@ class ModelManager:
149
  # prompt_list = [source_caption, target_caption, instruction]
150
  return image_links[0], image_links[1], image_links[2], prompt_list[0], prompt_list[1], prompt_list[2], model_names[0], model_names[1]
151
 
152
- @spaces.GPU(duration=200)
153
  def generate_video_vg(self, prompt, model_name):
154
  pipe = self.load_model_pipe(model_name)
155
  result = pipe(prompt=prompt)
 
149
  # prompt_list = [source_caption, target_caption, instruction]
150
  return image_links[0], image_links[1], image_links[2], prompt_list[0], prompt_list[1], prompt_list[2], model_names[0], model_names[1]
151
 
152
+ @spaces.GPU(duration=150)
153
  def generate_video_vg(self, prompt, model_name):
154
  pipe = self.load_model_pipe(model_name)
155
  result = pipe(prompt=prompt)