Skip to content

Commit

Permalink
Add mistral-7b-instruct alias
Browse files Browse the repository at this point in the history
  • Loading branch information
GregoryComer committed Apr 17, 2024
1 parent dc728a3 commit 4f56a24
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion build/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,11 @@ def from_name(cls, name: str):

# Aliases for well-known models. Maps a short name to a HuggingFace path. These
# can be used from the CLI in-place of the full model path.
model_aliases = {"llama2": "meta-llama/Llama-2-7b-chat-hf"}
model_aliases = {
"llama2": "meta-llama/Llama-2-7b-chat-hf",
"llama2-7": "meta-llama/Llama-2-7b-chat-hf",
"mistral-7b-instruct": "mistralai/Mistral-7B-Instruct-v0.2"
}

transformer_configs = {
"CodeLlama-7b-Python-hf": {
Expand Down

0 comments on commit 4f56a24

Please sign in to comment.