test.txt 15.3 KB
Newer Older
1
# This file was autogenerated by uv via the following command:
2
#    uv pip compile requirements/test.in -o requirements/test.txt --index-strategy unsafe-best-match --torch-backend cu128
3
4
5
6
7
8
9
10
absl-py==2.1.0
    # via rouge-score
accelerate==1.0.1
    # via
    #   lm-eval
    #   peft
aiohappyeyeballs==2.4.3
    # via aiohttp
11
aiohttp==3.10.11
12
13
14
15
16
17
18
19
20
21
22
    # via
    #   datasets
    #   fsspec
    #   lm-eval
aiosignal==1.3.1
    # via
    #   aiohttp
    #   ray
annotated-types==0.7.0
    # via pydantic
anyio==4.6.2.post1
23
24
25
    # via
    #   httpx
    #   starlette
26
27
argcomplete==3.5.1
    # via datamodel-code-generator
28
29
arrow==1.3.0
    # via isoduration
30
31
32
attrs==24.2.0
    # via
    #   aiohttp
33
    #   hypothesis
34
35
    #   jsonlines
    #   jsonschema
36
    #   pytest-subtests
37
38
39
    #   referencing
audioread==3.0.1
    # via librosa
40
backoff==2.2.1
41
42
43
    # via
    #   -r requirements/test.in
    #   schemathesis
Michael Goin's avatar
Michael Goin committed
44
bitsandbytes==0.45.3
45
    # via -r requirements/test.in
46
47
black==24.10.0
    # via datamodel-code-generator
48
49
blobfile==3.0.0
    # via -r requirements/test.in
50
boto3==1.35.57
51
    # via tensorizer
52
botocore==1.35.57
53
54
55
    # via
    #   boto3
    #   s3transfer
56
57
bounded-pool-executor==0.0.3
    # via pqdm
58
buildkite-test-collector==0.1.9
59
    # via -r requirements/test.in
60
61
62
63
64
65
66
67
68
69
70
71
72
73
certifi==2024.8.30
    # via
    #   httpcore
    #   httpx
    #   requests
cffi==1.17.1
    # via soundfile
chardet==5.2.0
    # via mbstrdecoder
charset-normalizer==3.4.0
    # via requests
click==8.1.7
    # via
    #   black
74
    #   jiwer
75
76
    #   nltk
    #   ray
77
    #   schemathesis
78
    #   typer
79
80
81
colorama==0.4.6
    # via
    #   sacrebleu
82
    #   schemathesis
83
84
85
    #   tqdm-multiprocess
contourpy==1.3.0
    # via matplotlib
86
87
cramjam==2.9.0
    # via fastparquet
88
89
90
91
cupy-cuda12x==13.3.0
    # via ray
cycler==0.12.1
    # via matplotlib
92
datamodel-code-generator==0.26.3
93
    # via -r requirements/test.in
94
95
96
97
98
99
100
101
dataproperty==1.0.1
    # via
    #   pytablewriter
    #   tabledata
datasets==3.0.2
    # via
    #   evaluate
    #   lm-eval
102
    #   mteb
103
104
105
106
107
108
109
110
111
112
decorator==5.1.1
    # via librosa
dill==0.3.8
    # via
    #   datasets
    #   evaluate
    #   lm-eval
    #   multiprocess
dnspython==2.7.0
    # via email-validator
113
114
docopt==0.6.2
    # via num2words
115
einops==0.8.0
116
    # via
117
    #   -r requirements/test.in
118
    #   encodec
Shinichi Hemmi's avatar
Shinichi Hemmi committed
119
    #   mamba-ssm
120
121
122
123
    #   vector-quantize-pytorch
    #   vocos
einx==0.3.0
    # via vector-quantize-pytorch
124
125
email-validator==2.2.0
    # via pydantic
126
127
encodec==0.1.1
    # via vocos
128
129
eval-type-backport==0.2.2
    # via mteb
130
131
evaluate==0.4.3
    # via lm-eval
132
133
fastparquet==2024.11.0
    # via genai-perf
134
135
fastrlock==0.8.2
    # via cupy-cuda12x
136
137
fastsafetensors==0.1.10
    # via -r requirements/test.in
138
139
filelock==3.16.1
    # via
140
    #   blobfile
141
142
143
144
145
146
147
    #   datasets
    #   huggingface-hub
    #   ray
    #   torch
    #   transformers
fonttools==4.54.1
    # via matplotlib
148
149
fqdn==1.5.1
    # via jsonschema
150
151
frozendict==2.4.6
    # via einx
152
153
154
155
156
frozenlist==1.5.0
    # via
    #   aiohttp
    #   aiosignal
    #   ray
157
fsspec==2024.9.0
158
159
160
    # via
    #   datasets
    #   evaluate
161
    #   fastparquet
162
163
    #   huggingface-hub
    #   torch
164
genai-perf==0.0.8
165
    # via -r requirements/test.in
166
167
genson==1.3.0
    # via datamodel-code-generator
168
169
graphql-core==3.2.6
    # via hypothesis-graphql
170
171
h11==0.14.0
    # via httpcore
172
173
harfile==0.3.0
    # via schemathesis
174
175
hf-xet==0.1.4
    # via huggingface-hub
176
177
178
179
180
hiredis==3.0.0
    # via tensorizer
httpcore==1.0.6
    # via httpx
httpx==0.27.2
181
182
183
    # via
    #   -r requirements/test.in
    #   schemathesis
184
huggingface-hub==0.30.1
185
    # via
186
    #   -r requirements/test.in
187
188
189
190
191
192
193
194
    #   accelerate
    #   datasets
    #   evaluate
    #   peft
    #   sentence-transformers
    #   timm
    #   tokenizers
    #   transformers
195
    #   vocos
196
197
humanize==4.11.0
    # via runai-model-streamer
198
199
200
201
202
203
204
205
206
hypothesis==6.131.0
    # via
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
hypothesis-graphql==0.11.1
    # via schemathesis
hypothesis-jsonschema==0.23.1
    # via schemathesis
207
208
209
210
211
idna==3.10
    # via
    #   anyio
    #   email-validator
    #   httpx
212
    #   jsonschema
213
214
215
216
217
218
    #   requests
    #   yarl
inflect==5.6.2
    # via datamodel-code-generator
iniconfig==2.0.0
    # via pytest
219
220
isoduration==20.11.0
    # via jsonschema
221
222
isort==5.13.2
    # via datamodel-code-generator
223
jinja2==3.1.6
224
225
226
    # via
    #   datamodel-code-generator
    #   torch
227
jiwer==3.0.5
228
    # via -r requirements/test.in
229
230
231
232
233
234
235
236
237
238
239
jmespath==1.0.1
    # via
    #   boto3
    #   botocore
joblib==1.4.2
    # via
    #   librosa
    #   nltk
    #   scikit-learn
jsonlines==4.0.0
    # via lm-eval
240
241
jsonpointer==3.0.0
    # via jsonschema
242
jsonschema==4.23.0
243
    # via
244
    #   hypothesis-jsonschema
245
246
    #   mistral-common
    #   ray
247
    #   schemathesis
248
249
jsonschema-specifications==2024.10.1
    # via jsonschema
250
251
junit-xml==1.9
    # via schemathesis
252
253
kaleido==0.2.1
    # via genai-perf
254
255
256
257
258
259
260
kiwisolver==1.4.7
    # via matplotlib
lazy-loader==0.4
    # via librosa
libnacl==2.1.0
    # via tensorizer
librosa==0.10.2.post1
261
    # via -r requirements/test.in
262
llvmlite==0.44.0
263
    # via numba
264
lm-eval==0.4.8
265
    # via -r requirements/test.in
266
lxml==5.3.0
267
268
269
    # via
    #   blobfile
    #   sacrebleu
Shinichi Hemmi's avatar
Shinichi Hemmi committed
270
271
mamba-ssm==2.2.4
    # via -r requirements/test.in
272
273
markdown-it-py==3.0.0
    # via rich
Huy Do's avatar
Huy Do committed
274
markupsafe==3.0.1
275
276
277
    # via
    #   jinja2
    #   werkzeug
278
matplotlib==3.9.2
279
    # via -r requirements/test.in
280
281
282
283
284
mbstrdecoder==1.1.3
    # via
    #   dataproperty
    #   pytablewriter
    #   typepy
285
286
mdurl==0.1.2
    # via markdown-it-py
287
mistral-common==1.5.4
288
    # via -r requirements/test.in
289
290
291
292
293
294
295
296
more-itertools==10.5.0
    # via lm-eval
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via
    #   librosa
    #   ray
297
298
mteb==1.38.11
    # via -r requirements/test.in
299
300
301
302
303
304
305
306
307
308
309
310
multidict==6.1.0
    # via
    #   aiohttp
    #   yarl
multiprocess==0.70.16
    # via
    #   datasets
    #   evaluate
mypy-extensions==1.0.0
    # via black
networkx==3.2.1
    # via torch
Shinichi Hemmi's avatar
Shinichi Hemmi committed
311
312
ninja==1.11.1.3
    # via mamba-ssm
313
314
nltk==3.9.1
    # via rouge-score
315
316
num2words==0.5.14
    # via -r requirements/test.in
cyyever's avatar
cyyever committed
317
numba==0.61.2
318
319
320
    # via
    #   -r requirements/test.in
    #   librosa
321
322
323
324
numexpr==2.10.1
    # via lm-eval
numpy==1.26.4
    # via
325
    #   -r requirements/test.in
326
327
328
329
330
    #   accelerate
    #   bitsandbytes
    #   contourpy
    #   cupy-cuda12x
    #   datasets
331
332
    #   einx
    #   encodec
333
    #   evaluate
334
335
    #   fastparquet
    #   genai-perf
336
337
    #   librosa
    #   matplotlib
338
    #   mistral-common
339
    #   mteb
340
341
    #   numba
    #   numexpr
342
    #   opencv-python-headless
343
    #   pandas
344
    #   patsy
345
346
    #   peft
    #   rouge-score
347
    #   runai-model-streamer
348
349
350
351
    #   sacrebleu
    #   scikit-learn
    #   scipy
    #   soxr
352
    #   statsmodels
353
354
355
    #   tensorizer
    #   torchvision
    #   transformers
356
    #   tritonclient
357
    #   vocos
358
nvidia-cublas-cu12==12.8.3.14
359
360
361
362
    # via
    #   nvidia-cudnn-cu12
    #   nvidia-cusolver-cu12
    #   torch
363
nvidia-cuda-cupti-cu12==12.8.57
364
    # via torch
365
nvidia-cuda-nvrtc-cu12==12.8.61
366
    # via torch
367
nvidia-cuda-runtime-cu12==12.8.57
368
    # via torch
369
nvidia-cudnn-cu12==9.7.1.26
370
    # via torch
371
nvidia-cufft-cu12==11.3.3.41
372
    # via torch
373
nvidia-cufile-cu12==1.13.0.11
374
    # via torch
375
nvidia-curand-cu12==10.3.9.55
376
    # via torch
377
nvidia-cusolver-cu12==11.7.2.55
Huy Do's avatar
Huy Do committed
378
    # via torch
379
nvidia-cusparse-cu12==12.5.7.53
380
381
382
    # via
    #   nvidia-cusolver-cu12
    #   torch
Huy Do's avatar
Huy Do committed
383
nvidia-cusparselt-cu12==0.6.3
Michael Goin's avatar
Michael Goin committed
384
    # via torch
Huy Do's avatar
Huy Do committed
385
nvidia-nccl-cu12==2.26.2
386
    # via torch
387
nvidia-nvjitlink-cu12==12.8.61
388
    # via
Huy Do's avatar
Huy Do committed
389
    #   nvidia-cufft-cu12
390
391
392
    #   nvidia-cusolver-cu12
    #   nvidia-cusparse-cu12
    #   torch
393
nvidia-nvtx-cu12==12.8.55
394
    # via torch
395
396
397
398
opencv-python-headless==4.11.0.86
    # via
    #   -r requirements/test.in
    #   mistral-common
399
packaging==24.2
400
401
402
403
404
405
    # via
    #   accelerate
    #   black
    #   datamodel-code-generator
    #   datasets
    #   evaluate
406
    #   fastparquet
407
408
    #   huggingface-hub
    #   lazy-loader
Shinichi Hemmi's avatar
Shinichi Hemmi committed
409
    #   mamba-ssm
410
411
    #   matplotlib
    #   peft
412
    #   plotly
413
414
415
416
    #   pooch
    #   pytest
    #   pytest-rerunfailures
    #   ray
417
    #   statsmodels
418
419
420
421
422
423
    #   transformers
    #   typepy
pandas==2.2.3
    # via
    #   datasets
    #   evaluate
424
425
426
    #   fastparquet
    #   genai-perf
    #   statsmodels
427
428
429
430
pathspec==0.12.1
    # via black
pathvalidate==3.2.1
    # via pytablewriter
431
432
patsy==1.0.1
    # via statsmodels
433
434
peft==0.13.2
    # via
435
    #   -r requirements/test.in
436
    #   lm-eval
437
pillow==10.4.0
438
    # via
439
    #   genai-perf
440
    #   matplotlib
441
    #   mistral-common
442
443
444
445
446
447
    #   sentence-transformers
    #   torchvision
platformdirs==4.3.6
    # via
    #   black
    #   pooch
448
449
plotly==5.24.1
    # via genai-perf
450
451
pluggy==1.5.0
    # via pytest
452
453
polars==1.29.0
    # via mteb
454
455
456
457
pooch==1.8.2
    # via librosa
portalocker==2.10.1
    # via sacrebleu
458
pqdm==0.2.0
459
    # via -r requirements/test.in
460
461
462
463
464
465
466
467
468
469
470
471
472
473
propcache==0.2.0
    # via yarl
protobuf==5.28.3
    # via
    #   ray
    #   tensorizer
psutil==6.1.0
    # via
    #   accelerate
    #   peft
    #   tensorizer
py==1.11.0
    # via pytest-forked
pyarrow==18.0.0
474
475
476
    # via
    #   datasets
    #   genai-perf
477
478
479
480
pybind11==2.13.6
    # via lm-eval
pycparser==2.22
    # via cffi
481
482
pycryptodomex==3.22.0
    # via blobfile
483
pydantic==2.9.2
484
485
486
    # via
    #   datamodel-code-generator
    #   mistral-common
487
    #   mteb
488
489
pydantic-core==2.23.4
    # via pydantic
490
491
pygments==2.18.0
    # via rich
492
493
pyparsing==3.2.0
    # via matplotlib
494
495
pyrate-limiter==3.7.0
    # via schemathesis
496
497
498
499
pytablewriter==1.2.0
    # via lm-eval
pytest==8.3.3
    # via
500
    #   -r requirements/test.in
501
    #   buildkite-test-collector
502
    #   genai-perf
503
504
    #   pytest-asyncio
    #   pytest-forked
505
    #   pytest-mock
506
507
    #   pytest-rerunfailures
    #   pytest-shard
508
    #   pytest-subtests
509
    #   pytest-timeout
510
    #   schemathesis
511
pytest-asyncio==0.24.0
512
    # via -r requirements/test.in
513
pytest-forked==1.6.0
514
    # via -r requirements/test.in
515
516
pytest-mock==3.14.0
    # via genai-perf
517
pytest-rerunfailures==14.0
518
    # via -r requirements/test.in
519
pytest-shard==0.1.2
520
    # via -r requirements/test.in
521
522
pytest-subtests==0.14.1
    # via schemathesis
523
524
pytest-timeout==2.3.1
    # via -r requirements/test.in
525
526
python-dateutil==2.9.0.post0
    # via
527
    #   arrow
528
529
530
531
    #   botocore
    #   matplotlib
    #   pandas
    #   typepy
532
533
python-rapidjson==1.20
    # via tritonclient
534
535
pytrec-eval-terrier==0.5.7
    # via mteb
536
537
538
539
540
541
542
543
544
pytz==2024.2
    # via
    #   pandas
    #   typepy
pyyaml==6.0.2
    # via
    #   accelerate
    #   datamodel-code-generator
    #   datasets
545
    #   genai-perf
546
547
548
    #   huggingface-hub
    #   peft
    #   ray
549
    #   responses
550
    #   schemathesis
551
552
    #   timm
    #   transformers
553
    #   vocos
554
555
rapidfuzz==3.12.1
    # via jiwer
Rui Qiao's avatar
Rui Qiao committed
556
ray==2.43.0
557
    # via -r requirements/test.in
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
redis==5.2.0
    # via tensorizer
referencing==0.35.1
    # via
    #   jsonschema
    #   jsonschema-specifications
regex==2024.9.11
    # via
    #   nltk
    #   sacrebleu
    #   tiktoken
    #   transformers
requests==2.32.3
    # via
    #   buildkite-test-collector
    #   datasets
    #   evaluate
    #   huggingface-hub
    #   lm-eval
577
    #   mistral-common
578
    #   mteb
579
580
    #   pooch
    #   ray
581
    #   responses
582
583
    #   schemathesis
    #   starlette-testclient
584
585
    #   tiktoken
    #   transformers
586
587
responses==0.25.3
    # via genai-perf
588
589
590
591
rfc3339-validator==0.1.4
    # via jsonschema
rfc3987==1.3.8
    # via jsonschema
592
rich==13.9.4
593
594
    # via
    #   genai-perf
595
    #   mteb
596
    #   typer
597
598
rouge-score==0.1.2
    # via lm-eval
599
rpds-py==0.20.1
600
601
602
    # via
    #   jsonschema
    #   referencing
603
runai-model-streamer==0.11.0
604
    # via -r requirements/test.in
605
runai-model-streamer-s3==0.11.0
606
    # via -r requirements/test.in
607
s3transfer==0.10.3
608
    # via boto3
609
610
611
612
613
614
615
616
sacrebleu==2.4.3
    # via lm-eval
safetensors==0.4.5
    # via
    #   accelerate
    #   peft
    #   timm
    #   transformers
617
618
schemathesis==3.39.15
    # via -r requirements/test.in
619
620
621
622
scikit-learn==1.5.2
    # via
    #   librosa
    #   lm-eval
623
    #   mteb
624
625
626
627
    #   sentence-transformers
scipy==1.13.1
    # via
    #   librosa
628
    #   mteb
629
630
    #   scikit-learn
    #   sentence-transformers
631
    #   statsmodels
632
    #   vocos
633
sentence-transformers==3.2.1
634
635
636
    # via
    #   -r requirements/test.in
    #   mteb
637
638
sentencepiece==0.2.0
    # via mistral-common
639
setuptools==77.0.3
640
    # via
Shinichi Hemmi's avatar
Shinichi Hemmi committed
641
    #   mamba-ssm
642
643
    #   pytablewriter
    #   torch
Huy Do's avatar
Huy Do committed
644
    #   triton
645
646
shellingham==1.5.4
    # via typer
647
648
six==1.16.0
    # via
649
    #   junit-xml
650
    #   python-dateutil
651
    #   rfc3339-validator
652
653
654
655
656
    #   rouge-score
sniffio==1.3.1
    # via
    #   anyio
    #   httpx
657
658
sortedcontainers==2.4.0
    # via hypothesis
659
660
soundfile==0.12.1
    # via
661
    #   -r requirements/test.in
662
663
664
665
666
    #   librosa
soxr==0.5.0.post1
    # via librosa
sqlitedict==2.1.0
    # via lm-eval
667
668
669
670
671
672
starlette==0.46.2
    # via
    #   schemathesis
    #   starlette-testclient
starlette-testclient==0.4.1
    # via schemathesis
673
674
statsmodels==0.14.4
    # via genai-perf
Huy Do's avatar
Huy Do committed
675
sympy==1.13.3
676
677
678
    # via
    #   einx
    #   torch
679
680
681
682
683
684
685
tabledata==1.3.3
    # via pytablewriter
tabulate==0.9.0
    # via sacrebleu
tcolorpy==0.1.6
    # via pytablewriter
tenacity==9.0.0
686
687
688
    # via
    #   lm-eval
    #   plotly
689
tensorizer==2.9.0
690
    # via -r requirements/test.in
691
692
threadpoolctl==3.5.0
    # via scikit-learn
693
694
695
696
tiktoken==0.7.0
    # via
    #   lm-eval
    #   mistral-common
697
timm==1.0.11
698
    # via -r requirements/test.in
699
700
701
702
tokenizers==0.21.1
    # via
    #   -r requirements/test.in
    #   transformers
703
704
705
706
tomli==2.2.1
    # via schemathesis
tomli-w==1.2.0
    # via schemathesis
707
torch==2.7.0+cu128
708
    # via
709
    #   -r requirements/test.in
710
711
    #   accelerate
    #   bitsandbytes
712
    #   encodec
713
    #   fastsafetensors
714
    #   lm-eval
Shinichi Hemmi's avatar
Shinichi Hemmi committed
715
    #   mamba-ssm
716
    #   mteb
717
    #   peft
718
    #   runai-model-streamer
719
720
721
    #   sentence-transformers
    #   tensorizer
    #   timm
722
    #   torchaudio
723
    #   torchvision
724
725
    #   vector-quantize-pytorch
    #   vocos
726
torchaudio==2.7.0+cu128
727
    # via
728
    #   -r requirements/test.in
729
730
    #   encodec
    #   vocos
731
torchvision==0.22.0+cu128
Michael Goin's avatar
Michael Goin committed
732
733
734
    # via
    #   -r requirements/test.in
    #   timm
735
736
737
738
739
740
tqdm==4.66.6
    # via
    #   datasets
    #   evaluate
    #   huggingface-hub
    #   lm-eval
741
    #   mteb
742
743
    #   nltk
    #   peft
744
    #   pqdm
745
746
747
748
749
    #   sentence-transformers
    #   tqdm-multiprocess
    #   transformers
tqdm-multiprocess==0.0.11
    # via lm-eval
750
transformers==4.51.3
751
    # via
752
    #   -r requirements/test.in
753
    #   genai-perf
754
    #   lm-eval
Shinichi Hemmi's avatar
Shinichi Hemmi committed
755
    #   mamba-ssm
756
757
758
759
    #   peft
    #   sentence-transformers
    #   transformers-stream-generator
transformers-stream-generator==0.0.5
760
    # via -r requirements/test.in
Huy Do's avatar
Huy Do committed
761
triton==3.3.0
762
    # via torch
763
764
tritonclient==2.51.0
    # via
765
    #   -r requirements/test.in
766
    #   genai-perf
767
typepy==1.3.2
768
769
770
771
    # via
    #   dataproperty
    #   pytablewriter
    #   tabledata
772
773
typer==0.15.2
    # via fastsafetensors
774
775
types-python-dateutil==2.9.0.20241206
    # via arrow
776
777
778
779
typing-extensions==4.12.2
    # via
    #   huggingface-hub
    #   librosa
780
    #   mistral-common
781
    #   mteb
782
    #   pqdm
783
784
785
    #   pydantic
    #   pydantic-core
    #   torch
786
    #   typer
787
788
tzdata==2024.2
    # via pandas
789
790
uri-template==1.3.0
    # via jsonschema
791
urllib3==2.2.3
792
    # via
793
    #   blobfile
794
795
    #   botocore
    #   requests
796
797
    #   responses
    #   tritonclient
798
vector-quantize-pytorch==1.21.2
799
    # via -r requirements/test.in
800
vocos==0.1.0
801
    # via -r requirements/test.in
802
803
804
805
webcolors==24.11.1
    # via jsonschema
werkzeug==3.1.3
    # via schemathesis
806
807
808
809
810
811
word2number==1.1
    # via lm-eval
xxhash==3.5.0
    # via
    #   datasets
    #   evaluate
812
yarl==1.17.1
813
814
815
    # via
    #   aiohttp
    #   schemathesis
816
817
zstandard==0.23.0
    # via lm-eval