wehub-resource-sync
83 lines
2.9 KiB
Plaintext
83 lines
2.9 KiB
Plaintext
-r ../common.txt
|
|
|
|
# testing
|
|
pytest
|
|
tensorizer==2.10.1
|
|
pytest-forked
|
|
pytest-asyncio
|
|
pytest-rerunfailures
|
|
pytest-shard
|
|
pytest-timeout
|
|
pytest-cov
|
|
|
|
# testing utils
|
|
albumentations # required for Nemotron Parse in test_common.py
|
|
av # required for audio_in_video tests
|
|
torchcodec >= 0.14 # required for torchcodec video backend tests
|
|
backoff # required for phi4mm test
|
|
blobfile # required for kimi-vl test
|
|
httpx
|
|
librosa # required for audio tests
|
|
vector_quantize_pytorch # required for minicpmo_26 test
|
|
vocos # required for minicpmo_26 test
|
|
peft>=0.19.1 # required for phi-4-mm test
|
|
pqdm
|
|
ray[cgraph,default]>=2.48.0 # Ray Compiled Graph, required by pipeline parallelism tests
|
|
sentence-transformers>=5.2.0 # required for embedding tests
|
|
soundfile # required for audio tests
|
|
jiwer # required for audio tests
|
|
tblib # for pickling test exceptions
|
|
timm >=1.0.17 # required for internvl and gemma3n-mm test
|
|
torch==2.11.0
|
|
torchaudio==2.11.0
|
|
torchvision==0.26.0
|
|
transformers_stream_generator # required for qwen-vl test
|
|
matplotlib # required for qwen-vl test
|
|
mistral_common[image,audio] >= 1.11.5 # required for voxtral test
|
|
num2words # required for smolvlm test
|
|
open_clip_torch==2.32.0 # Required for nemotron_vl test, Nemotron Parse in test_common.py
|
|
datamodel_code_generator # required for minicpm3 test
|
|
lm-eval[api]>=0.4.12 # required for model evaluation test
|
|
mteb[bm25s]>=2, <3 # required for mteb test
|
|
transformers==5.13.1
|
|
tokenizers==0.22.2
|
|
schemathesis>=4.0.0 # Required for openai schema test.
|
|
# quantization
|
|
bitsandbytes==0.49.2
|
|
buildkite-test-collector==0.1.9
|
|
|
|
|
|
genai_perf>=0.0.8
|
|
tritonclient>=2.51.0
|
|
|
|
# The version of gRPC libraries should be consistent with each other
|
|
grpcio==1.78.0
|
|
grpcio-reflection==1.78.0
|
|
|
|
arctic-inference == 0.1.1; platform_machine == "x86_64" # Required for suffix decoding test
|
|
numba == 0.65.0 # Required for N-gram speculative decoding
|
|
runai-model-streamer[s3,gcs,azure]==0.15.7
|
|
fastsafetensors>=0.3.2
|
|
instanttensor>=0.1.5; platform_machine == "x86_64"
|
|
decord==0.6.0; platform_machine == "x86_64"
|
|
# terratorch is temporarily disabled while PyPI has the `lightning` package
|
|
# in `quarantined` status (every published terratorch version transitively
|
|
# requires `lightning`, so the resolver fails with "no versions of lightning").
|
|
# Re-enable once PyPI lifts the quarantine. Tracked in #41376.
|
|
# terratorch >= 1.2.2 # Required for Prithvi tests
|
|
imagehash # Required for Prithvi tests
|
|
segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests
|
|
|
|
gpt-oss >= 0.0.7; python_version > '3.11'
|
|
|
|
perceptron # required for isaac test
|
|
kaldi-native-fbank >= 1.18.7 # required for fireredasr2 test
|
|
cohere_melody>=0.9.0 # required for cohere command reasoning parser test
|
|
|
|
# Newer versions of datasets require torchcoded, that makes the tests fail in CI because of a missing library.
|
|
# Older versions are in conflict with teerratorch requirements.
|
|
datasets>=3.3.0,<=3.6.0
|
|
|
|
openpyxl # required for perf comparison excel report
|
|
plotly # required for perf comparison html report
|