Hugging Face's logo

Spaces:
fywalter
/
nudging_align
Sleeping

App Files Community
fywalter commited on
Commit
5c8369d
•
1 Parent(s): bc473cb

fix model name bug

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -11,7 +11,7 @@ NUM_LOGPROBS = {
11
  MODEL_MAPPING = {
12
  "Llama-2-70B": "meta-llama/Llama-2-70b-hf",
13
  "Mistral-7B-v0.1": "mistralai/Mistral-7B-v0.1",
14
- "Mixtral-8x7B": "mistralai/Mixtral-8x7B-v0.1",
15
  # Nudging models below
16
  "Mistral-7B-v0.1-Instruct": "mistralai/Mistral-7B-Instruct-v0.1",
17
  "Llama-2-13B-chat": "meta-llama/Llama-2-13b-chat-hf",
 
11
  MODEL_MAPPING = {
12
  "Llama-2-70B": "meta-llama/Llama-2-70b-hf",
13
  "Mistral-7B-v0.1": "mistralai/Mistral-7B-v0.1",
14
+ "Mixtral-8x7B-v0.1": "mistralai/Mixtral-8x7B-v0.1",
15
  # Nudging models below
16
  "Mistral-7B-v0.1-Instruct": "mistralai/Mistral-7B-Instruct-v0.1",
17
  "Llama-2-13B-chat": "meta-llama/Llama-2-13b-chat-hf",