Add Mistral 7B to `test_models` (#1366)

95248677 · Woosuk Kwon · GitHub · c1376e0f · 95248677
Unverified Commit 95248677 authored Oct 16, 2023 by Woosuk Kwon Committed by GitHub Oct 16, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

tests/models/test_models.py tests/models/test_models.py +3 -2

No files found.
--- a/tests/models/test_models.py
+++ b/tests/models/test_models.py
@@ -6,14 +6,15 @@ import pytest

 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",
 ]