From 95248677014cb10a9dbaa2e72f688e1a6e6cf566 Mon Sep 17 00:00:00 2001
From: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Date: Mon, 16 Oct 2023 17:49:54 -0700
Subject: [PATCH] Add Mistral 7B to `test_models` (#1366)

---
 tests/models/test_models.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/models/test_models.py b/tests/models/test_models.py
index 92949a8064147..8b059ad5e0a45 100644
--- 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",
 ]