diff --git a/requirements-test.txt b/requirements-test.txt index 62d6cc49eade4..467545ca07c0e 100644 --- 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