[FEATURE] Include Mistral-7B model in list of supported base models (#…

…3674)
ludwig-ai · Sep 28, 2023 · f87376b · f87376b
1 parent 3dc8f4b
commit f87376b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",
 }