llava-1-5 / configs.py
GDinesh's picture
Update configs.py
adf37d6 verified
raw
history blame contribute delete
568 Bytes
MODEL_PATH = "GDinesh/llava-300k-lora-unmerged-21-mar"# "./llava-v1.5-13b-300k-checkpoint"#"saurabh-straive/llava-v1.5-13b-task-lora-100k" #"checkpoints/llava-v1.5-13b-task-lora-100k" #"liuhaotian/llava-v1.5-13b" #"liuhaotian/llava-llama-2-13b-chat-lightning-preview" #"daryl149/llama-2-7b-chat-hf" #"meta-llama/Llama-2-13b-chat-hf"
CONV_MODE = "llava_llama_2"
MODEL_BASE = "liuhaotian/llava-v1.5-13b" #None
MODEL_NAME = "llava" #"llava-v1.5-13b-lora-finetuned"
LOAD_8BIT = False
LOAD_4BIT = False
IMAGE_ASPECT_RATIO = "pad"
TEMPERATURE = 0.01
MAX_NEW_TOKENS = 1024