KB-VQA / my_model /config /kbvqa_config.py
m7mdal7aj's picture
Update my_model/config/kbvqa_config.py
1a0e500 verified
raw
history blame
1.15 kB
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
LOW_CPU_MEM_USAGE = True
# Access Token
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
# Default SYS Prompt
SYSTEM_PROMPT = "You are a helpful, respectful and honest assistant for visual question answering. you are provided with a caption of an image and a list of objects detected in the image along with their bounding boxes and level of certainty, you will output an answer to the given questions in no more than one sentence. Use logical reasoning to reach to the answer, but do not output your reasoning process unless asked for it. If provided, you will use the [CAP] and [/CAP] tags to indicate the begining and end of the caption respectively. If provided you will use the [OBJ] and [/OBJ] tags to indicate the begining and end of the list of detected objects in the image along with their bounding boxes respectively.if provided, you will use [QES] and [/QES] tags to indicate the begining and end of the question respectively."