Update mistral tests to fully open source version. (deepspeedai#476)

fkatada · May 15, 2024 · dcc4190 · dcc4190
1 parent 56af8d0
commit dcc4190
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/requirements/requirements-dev.txt b/requirements/requirements-dev.txt
@@ -2,3 +2,4 @@ clang-format==16.0.2
 pre-commit>=2.20.0
 pytest
 pytest-forked
+sentencepiece
diff --git a/tests/test_model_support.py b/tests/test_model_support.py
@@ -82,7 +82,7 @@ def inject_checkpoint_engine():
     [
         "tiiuae/falcon-7b",
         "NousResearch/Llama-2-7b-hf",
-        "mistralai/Mistral-7B-v0.1",
+        "NousResearch/Hermes-2-Pro-Mistral-7B",
         "cloudyu/Mixtral_11Bx2_MoE_19B",
         "facebook/opt-125m",
     ],

diff --git a/tests/test_ragged_batching.py b/tests/test_ragged_batching.py
@@ -14,7 +14,7 @@
     [
         "tiiuae/falcon-7b",
         "NousResearch/Llama-2-7b-hf",
-        "mistralai/Mistral-7B-v0.1",
+        "NousResearch/Hermes-2-Pro-Mistral-7B",
         "cloudyu/Mixtral_11Bx2_MoE_19B",
         "facebook/opt-125m",
     ],