tonic commited on
Commit
91d8e22
2 Parent(s): bc016c4 b1062e5

Merge branch 'main' of https://huggingface.co/spaces/Tonic/Prometheus-VL

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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( model_path = model_path , model_name = model_name, image_folder, question_file, answers_file, conv_mode, num_chunks, chunk_idx, temperature, top_p, num_beams):
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