rocm-test.txt 904 Bytes
Newer Older
1
2
# Common dependencies
-r common.txt
3
tblib==3.1.0
4
5
bm25s==0.2.13
pystemmer==3.0.0
6

7
# Entrypoints test
8
9
10
11
12
13
14
15
16
17
18
19
# librosa==0.10.2.post1 # required by audio tests in entrypoints/openai
audioread==3.0.1
cffi==1.17.1
decorator==5.2.1
lazy-loader==0.4
platformdirs==4.3.6
pooch==1.8.2
#pycparse==2.22
soundfile==0.13.1
soxr==0.5.0.post1
librosa==0.10.2.post1

20
# Entrypoints test
21
22
23
#vllm[video] # required by entrypoints/openai/test_video.py
decord==0.6.0

24
# Entrypoints test
25
26
27
#sentence-transformers # required by entrypoints/openai/test_score.py
sentence-transformers==3.4.1

28
29
30
31
32
33
# Basic Models Test
matplotlib==3.10.3

# Multi-Modal Models Test (Extended) 3
blobfile==3.0.0

34
# Required for openai schema test.
35
schemathesis==3.39.15
36

37
38
39
40
41
# Required for mteb test
mteb[bm25s]>=1.38.11, <2

# Required for eval tests
lm-eval[api] @ git+https://github.com/EleutherAI/lm-evaluation-harness.git@206b7722158f58c35b7ffcd53b035fdbdda5126d