VICUNA_PATH = "/data3/public/model/vicuna-13b-v1.5"
LLAMA_PATH = "/mnt/public/gpfs-jd/code/guoweiyang/LRM/model/Llama2-7B-chat"

VICUNA_API_LINK ="https://..." # ADD LINK
LLAMA_API_LINK = "https://..." # ADD LINK


ATTACK_TEMP = 1
TARGET_TEMP = 0
ATTACK_TOP_P = 0.9
TARGET_TOP_P = 1

# Increase the above allow more streams in parallel
# Decrease it to reduce the memory requirement 
MAX_PARALLEL_STREAMS = 5
