ermu2001 commited on
Commit
25e34e0
1 Parent(s): 270e904

Update tasks/eval/model_utils.py

Browse files
Files changed (1) hide show
  1. tasks/eval/model_utils.py +1 -1
tasks/eval/model_utils.py CHANGED
@@ -137,7 +137,7 @@ def load_adapters(model, adapter_model_name_or_paths):
137
  return model
138
 
139
 
140
- @spaces.GPU(duration=120)
141
  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,
142
  repetition_penalty=1.0, length_penalty=1, temperature=1.0, stop_criteria_keywords=None, print_res=False):
143
  # torch.cuda.empty_cache()
 
137
  return model
138
 
139
 
140
+ @spaces.GPU(duration=30)
141
  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,
142
  repetition_penalty=1.0, length_penalty=1, temperature=1.0, stop_criteria_keywords=None, print_res=False):
143
  # torch.cuda.empty_cache()