KB-VQA-E / my_model /config /kbvqa_config.py
m7mdal7aj's picture
Create config/kbvqa_config.py
d8fd7b8 verified
raw
history blame
274 Bytes
import os
# Model and Tokenizer Settings
KBVQA_MODEL_NAME = "m7mdal7aj/fine_tunned_llama_2_merged"
QUANTIZATION = '4bit'
MAX_CONTEXT_WINDOW = 4000
ADD_EOS_TOKEN = False
TRUST_REMOTE = False
USE_FAST = True
# Access Token
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")