Add e5-mistral embedding model - step 3/3 (#988)

2024-08-08 16:31:19 -07:00
parent 9f662501a3
commit e040a2450b
14 changed files with 474 additions and 241 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -10,7 +10,8 @@ suites = {
        "test_vision_openai_server.py",
        "test_chunked_prefill.py",
        "test_torch_compile.py",
-        "models/test_causal_models.py",
+        "models/test_generation_models.py",
+        "models/test_embedding_models.py",
        "sampling/penaltylib",
    ],
    "sampling/penaltylib": glob.glob(