Change duration

model/model_manager.py

@@ -130,7 +130,7 @@ class ModelManager:
             # guard is disabled
             return "safe"
     
-    @spaces.GPU(duration=120)
+    @spaces.GPU(duration=180)
     def generate_image_ig(self, prompt, model_name):
         # if 'unsafe' not in self.NSFW_filter(prompt):
         print('The prompt is safe')
@@ -209,7 +209,7 @@ class ModelManager:
         return image_links[0], image_links[1], prompt_list[0]
 
 
-    @spaces.GPU(duration=200)
+    @spaces.GPU(duration=180)
     def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
         # if 'unsafe' not in self.NSFW_filter(" ".join([textbox_source, textbox_target, textbox_instruct])):
         pipe = self.load_model_pipe(model_name)
@@ -277,7 +277,7 @@ class ModelManager:
             # prompt_list = [source_caption, target_caption, instruction]
         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]
 
-    @spaces.GPU(duration=150)
+    @spaces.GPU(duration=180)
     def generate_video_vg(self, prompt, model_name):
         # if 'unsafe' not in self.NSFW_filter(prompt):
         pipe = self.load_model_pipe(model_name)