Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA-E
Running

App Files Community
m7mdal7aj commited on
Commit
ad362a6
•
1 Parent(s): e59b1eb

Update my_model/config/kbvqa_config.py

Browse files
Files changed (1) hide show
  1. my_model/config/kbvqa_config.py +2 -1
my_model/config/kbvqa_config.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
 
3
  # Model and Tokenizer Settings
4
- KBVQA_MODEL_NAME = "m7mdal7aj/fine_tunned_llama_2_merged"
 
5
  QUANTIZATION = '4bit'
6
  MAX_CONTEXT_WINDOW = 4000
7
  ADD_EOS_TOKEN = False
 
1
  import os
2
 
3
  # Model and Tokenizer Settings
4
+ KBVQA_MODEL_NAME_7b = "m7mdal7aj/fine_tuned_llama_2_7b_chat_OKVQA"
5
+ KBVQA_MODEL_NAME_13b = "m7mdal7aj/fine_tuned_llama_2_13b_chat_OKVQA"
6
  QUANTIZATION = '4bit'
7
  MAX_CONTEXT_WINDOW = 4000
8
  ADD_EOS_TOKEN = False