Add Mistral 7B to test_models (#1366)

vllm-project · Oct 17, 2023 · 9524867 · 9524867
1 parent c1376e0
commit 9524867
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/models/test_models.py b/tests/models/test_models.py
@@ -6,14 +6,15 @@
 
 MODELS = [
     "facebook/opt-125m",
+    "meta-llama/Llama-2-7b-hf",
+    "mistralai/Mistral-7B-v0.1",
+    "tiiuae/falcon-7b",
     "gpt2",
     "bigcode/tiny_starcoder_py",
     "EleutherAI/gpt-j-6b",
     "EleutherAI/pythia-70m",
     "bigscience/bloom-560m",
     "mosaicml/mpt-7b",
-    "tiiuae/falcon-7b",
-    "meta-llama/Llama-2-7b-hf",
 ]