Spaces:
Runtime error
Runtime error
Add wizard lm
Browse files- backend/query_llm.py +1 -0
backend/query_llm.py
CHANGED
@@ -14,6 +14,7 @@ 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"}
|
|
|
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 |
+
"WizardLM-30B": "cognitivecomputations/WizardLM-30B-Uncensored",
|
18 |
"llama 3": "meta-llama/Meta-Llama-3-70B-Instruct",
|
19 |
}
|
20 |
openai_models = {"gpt-4o","gpt-3.5-turbo-0125"}
|