Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
Commit
•
83e2394
1
Parent(s):
5f52571
Update model/model_manager.py
Browse files- 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=
|
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)
|