complynx commited on
Commit
d2208d6
Β·
1 Parent(s): 5dedbfc

Add vicuna

Browse files
Files changed (1) hide show
  1. backend/query_llm.py +1 -0
backend/query_llm.py CHANGED
@@ -13,6 +13,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
13
  hf_models = {
14
  "mistral-7B": "mistralai/Mistral-7B-Instruct-v0.2",
15
  "mistral-7B 0.1": "mistralai/Mistral-7B-v0.1",
 
16
  "llama 3": "meta-llama/Meta-Llama-3-70B-Instruct",
17
  }
18
  openai_models = {"gpt-4o","gpt-3.5-turbo-0125"}
 
13
  hf_models = {
14
  "mistral-7B": "mistralai/Mistral-7B-Instruct-v0.2",
15
  "mistral-7B 0.1": "mistralai/Mistral-7B-v0.1",
16
+ "vicuna-13b":"lmsys/vicuna-13b-v1.5",
17
  "llama 3": "meta-llama/Meta-Llama-3-70B-Instruct",
18
  }
19
  openai_models = {"gpt-4o","gpt-3.5-turbo-0125"}