[CI/Build] Add text-only test for Qwen models (#7475)

Signed-off-by: Alex-Brooks <Alex.Brooks@ibm.com>
This commit is contained in:
Alex Brooks
2024-08-18 17:43:46 -06:00
committed by GitHub
parent e3b318216d
commit 40e1360bb6
3 changed files with 52 additions and 2 deletions
+3 -1
View File
@@ -11,7 +11,7 @@ pytest-shard
# testing utils
awscli
einops # required for MPT
einops # required for MPT and qwen-vl
httpx
peft
requests
@@ -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 # required for qwen-vl test
matplotlib # required for qwen-vl test
# TODO: Add this after fully implementing llava(mantis)
# git+https://github.com/TIGER-AI-Lab/Mantis.git # required for llava(mantis) test