Spaces:
Runtime error
Runtime error
Merge branch 'main' of https://huggingface.co/spaces/Tonic/Prometheus-VL
Browse files- app.py +1 -1
- requirements.txt +2 -2
app.py
CHANGED
|
@@ -96,7 +96,7 @@ def eval_model(args, model_name = model_name, model_path = model_path):
|
|
| 96 |
ans_file.flush()
|
| 97 |
ans_file.close()
|
| 98 |
|
| 99 |
-
def gradio_wrapper(
|
| 100 |
|
| 101 |
question_file_path = os.path.join(tempfile.mkdtemp(), "question.jsonl")
|
| 102 |
with open(question_file_path, "w") as f:
|
|
|
|
| 96 |
ans_file.flush()
|
| 97 |
ans_file.close()
|
| 98 |
|
| 99 |
+
def gradio_wrapper( image_folder, question_file, answers_file, conv_mode, num_chunks, chunk_idx, temperature, top_p, num_beams, model_path = model_path , model_name = model_name):
|
| 100 |
|
| 101 |
question_file_path = os.path.join(tempfile.mkdtemp(), "question.jsonl")
|
| 102 |
with open(question_file_path, "w") as f:
|
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
torch
|
| 2 |
torchvision
|
| 3 |
-
flash-attn
|
| 4 |
transformers
|
| 5 |
accelerate
|
| 6 |
-
bitsandbytes
|
|
|
|
|
|
| 1 |
torch
|
| 2 |
torchvision
|
|
|
|
| 3 |
transformers
|
| 4 |
accelerate
|
| 5 |
+
bitsandbytes
|
| 6 |
+
shortuuid
|