Use smaller embedding model when not testing model specifically (#13891)

2025-02-28 08:50:43 +00:00
parent b9e41734c5
commit 76c89fcadd
9 changed files with 15 additions and 15 deletions
--- a/tests/entrypoints/llm/test_encode.py
+++ b/tests/entrypoints/llm/test_encode.py
@@ -8,7 +8,7 @@ import pytest
 from vllm import LLM, PoolingParams, PoolingRequestOutput
 from vllm.distributed import cleanup_dist_env_and_memory

-MODEL_NAME = "intfloat/e5-mistral-7b-instruct"
+MODEL_NAME = "intfloat/multilingual-e5-small"

 PROMPTS = [
    "Hello, my name is",