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