File size: 526 Bytes
8874c09
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
MODEL_PATH =  "./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