Spaces:
Runtime error
Runtime error
Add vicuna
Browse files- 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"}
|