diff --git a/ludwig/schema/llms/base_model.py b/ludwig/schema/llms/base_model.py index 305ef8971d0..7aff31718a0 100644 --- a/ludwig/schema/llms/base_model.py +++ b/ludwig/schema/llms/base_model.py @@ -31,6 +31,7 @@ "gpt-j-6b": "EleutherAI/gpt-j-6b", "pythia-2.8b": "EleutherAI/pythia-2.8b", "pythia-12b": "EleutherAI/pythia-12b", + "mistral-7b": "mistralai/Mistral-7B-v0.1", }