[Minor] Add Phi 2 to supported models (#2159)

2023-12-17 02:54:57 -08:00
parent c9fadda543
commit f8c688d746
3 changed files with 5 additions and 5 deletions
--- a/tests/models/test_models.py
+++ b/tests/models/test_models.py
@ -15,12 +15,12 @@ MODELS = [
    "EleutherAI/pythia-70m",
    "bigscience/bloom-560m",
    "mosaicml/mpt-7b",
-    "microsoft/phi-1_5",
+    "microsoft/phi-2",
 ]


@pytest.mark.parametrize("model", MODELS)
-@pytest.mark.parametrize("dtype", ["half"])
+@pytest.mark.parametrize("dtype", ["float"])
@pytest.mark.parametrize("max_tokens", [128])
 def test_models(
    hf_runner,