feat(model): Support Starling-LM-7B-beta (#1350)

This commit is contained in:
Fangyin Cheng
2024-03-29 17:25:30 +08:00
committed by GitHub
parent e0ab852a60
commit a272d1b8f3
5 changed files with 59 additions and 0 deletions

View File

@@ -156,6 +156,7 @@ LLM_MODEL_CONFIG = {
"gemma-7b-it": os.path.join(MODEL_PATH, "gemma-7b-it"),
# https://huggingface.co/google/gemma-2b-it
"gemma-2b-it": os.path.join(MODEL_PATH, "gemma-2b-it"),
"starling-lm-7b-beta": os.path.join(MODEL_PATH, "Starling-LM-7B-beta"),
}
EMBEDDING_MODEL_CONFIG = {