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

Add wizard lm

Browse files
Files changed (1) hide show
  1. 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"}