115 lines
2.5 KiB
Plaintext
115 lines
2.5 KiB
Plaintext
# Common dependencies
|
|
-r common.txt
|
|
|
|
# Test infrastructure
|
|
tblib==3.1.0
|
|
pytest==8.3.5
|
|
pytest-asyncio==0.24.0
|
|
pytest-timeout==2.3.1
|
|
pytest-cov==6.3.0
|
|
pytest-forked==1.6.0
|
|
pytest-rerunfailures==14.0
|
|
pytest-shard==0.1.2
|
|
|
|
# Async/HTTP dependencies
|
|
anyio==4.6.2.post1
|
|
# via httpx, starlette
|
|
aiohttp==3.13.3
|
|
# via gpt-oss
|
|
httpx==0.27.2
|
|
# HTTP testing
|
|
|
|
# Audio processing dependencies
|
|
audioread==3.0.1
|
|
# via librosa
|
|
cffi==1.17.1
|
|
# via soundfile
|
|
decorator==5.2.1
|
|
# via librosa
|
|
lazy-loader==0.4
|
|
# via librosa
|
|
platformdirs==4.3.6
|
|
# via pooch
|
|
pooch==1.8.2
|
|
# via librosa
|
|
soundfile==0.13.1
|
|
# via librosa
|
|
soxr==0.5.0.post1
|
|
# via librosa
|
|
librosa==0.10.2.post1
|
|
|
|
# Retrieval and search
|
|
bm25s==0.2.13
|
|
# via mteb
|
|
pystemmer==3.0.0
|
|
# via mteb
|
|
|
|
# Multi-modal processing
|
|
av==16.1.0
|
|
# required for audio_in_video tests
|
|
blobfile==3.0.0
|
|
# Multi-Modal Models Test
|
|
decord==0.6.0
|
|
# video processing, required by entrypoints/openai/chat_completion/test_video.py
|
|
rapidfuzz==3.12.1
|
|
|
|
# OpenAI compatibility and testing
|
|
gpt-oss==0.0.8
|
|
# OpenAI compatibility tests
|
|
schemathesis==3.39.15
|
|
# OpenAI schema test
|
|
|
|
# Evaluation and benchmarking
|
|
lm-eval[api]==0.4.11
|
|
jiwer==4.0.0
|
|
|
|
# Required for multiprocessed tests that use spawn method, Datasets and Evaluate Test
|
|
multiprocess==0.70.16
|
|
|
|
# Required for v1/metrics/test_engine_logger_apis.py
|
|
ray[cgraph,default]>=2.48.0
|
|
|
|
torchgeo==0.7.0
|
|
# via terratorch
|
|
# MTEB Benchmark Test
|
|
mteb[bm25s]>=2, <3
|
|
|
|
# Utilities
|
|
num2words==0.5.14
|
|
# via lm-eval
|
|
pqdm==0.2.0
|
|
# via lm-eval
|
|
|
|
# Required for fastsafetensors test
|
|
fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@d6f998a03432b2452f8de2bb5cefb5af9795d459
|
|
# Required for suffix decoding test
|
|
arctic-inference == 0.1.1
|
|
# Required for Nemotron test
|
|
open-clip-torch==2.32.0
|
|
# Required for isaac Multi-Modal generation test
|
|
perceptron==0.1.4
|
|
# Required for the multi-modal models test
|
|
timm==1.0.17
|
|
# Required for plugins test
|
|
albumentations==1.4.6
|
|
# Pin transformers version
|
|
transformers==4.57.5
|
|
# Pin HF Hub version
|
|
huggingface-hub==0.36.2
|
|
# Pin Mistral Common
|
|
mistral-common[image,audio]==1.10.0
|
|
# Required for Prithvi tests
|
|
terratorch==1.2.2
|
|
# Required for Prithvi tests
|
|
segmentation-models-pytorch==0.5.0
|
|
# Required for Prithvi tests
|
|
imagehash==4.3.2
|
|
# Required for bitsandbytes quantization test
|
|
bitsandbytes==0.49.2
|
|
# Examples (tensorizer) tests
|
|
tensorizer==2.10.1
|
|
# Multi-modal models test (`allendou/FireRedASR2-LLM-vllm`)
|
|
kaldi-native-fbank==1.22.3
|
|
# Pinning numpy version
|
|
numpy==2.2.6
|