nightly_torch_test.txt

# Dependency that able to run entrypoints test
# pytest and its extensions
pytest
pytest-asyncio
pytest-forked
pytest-mock
pytest-rerunfailures
pytest-shard
pytest-timeout

librosa # required by audio tests in entrypoints/openai
sentence-transformers
numba == 0.61.2; python_version > '3.9'
# testing utils
boto3
botocore
datasets
ray >= 2.10.0
peft
runai-model-streamer==0.11.0
runai-model-streamer-s3==0.11.0
tensorizer>=2.9.0
lm-eval==0.4.8
buildkite-test-collector==0.1.9
lm-eval[api]==0.4.8 # required for model evaluation test

# required for quantization test
bitsandbytes>=0.45.3

# required for minicpmo_26 test
vector_quantize_pytorch
vocos

# required for Basic Models Test
blobfile # required for kimi-vl test
matplotlib # required for qwen-vl test

# required for  Multi-Modal Models Test (Standard)
num2words # required for smolvlm test
pqdm
timm # required for internvl test