Add einops and transformers_stream_generator to test deps

Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
vllm-project · Aug 14, 2024 · 097acb8 · 097acb8
1 parent 611e37b
commit 097acb8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -19,6 +19,8 @@ ray
 sentence-transformers # required for embedding
 compressed-tensors==0.4.0 # required for compressed-tensors
 timm # required for internvl test
+transformers_stream_generator<1 # required for qwen test
+einops<1 # required for qwen test
 
 # TODO: Add this after fully implementing llava(mantis)
 # git+https://github.com/TIGER-AI-Lab/Mantis.git # required for llava(mantis) test