Update tasks/eval/model_utils.py

tasks/eval/model_utils.py

@@ -137,8 +137,8 @@ def load_adapters(model, adapter_model_name_or_paths):
     return model
 
 
-@spaces.GPU(duration=30)
-def pllava_answer(conv: Conversation, model, processor, img_list, do_sample=True, max_new_tokens=200, num_beams=1, min_length=1, top_p=0.9,
+@spaces.GPU(duration=200)
+def pllava_answer(conv: Conversation, model, processor, img_list, do_sample=True, max_new_tokens=400, num_beams=1, min_length=1, top_p=0.9,
                repetition_penalty=1.0, length_penalty=1, temperature=1.0, stop_criteria_keywords=None, print_res=False):
     # torch.cuda.empty_cache()
     prompt = conv.get_prompt()