Spaces:
Runtime error
Runtime error
Update tasks/eval/model_utils.py
Browse files
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=
|
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()
|