test.txt 12.9 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
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
23
24
25
26
27
28
29
30
31
32
33
    # via
    #   datasets
    #   fsspec
    #   lm-eval
aiosignal==1.3.1
    # via
    #   aiohttp
    #   ray
annotated-types==0.7.0
    # via pydantic
anyio==4.6.2.post1
    # via httpx
argcomplete==3.5.1
    # via datamodel-code-generator
attrs==24.2.0
    # via
    #   aiohttp
    #   jsonlines
    #   jsonschema
    #   referencing
audioread==3.0.1
    # via librosa
34
awscli==1.35.23
35
    # via -r requirements/test.in
36
37
backoff==2.2.1
    # via -r requirements/test.in
Michael Goin's avatar
Michael Goin committed
38
bitsandbytes==0.45.3
39
    # via -r requirements/test.in
40
41
black==24.10.0
    # via datamodel-code-generator
42
boto3==1.35.57
43
    # via tensorizer
44
botocore==1.35.57
45
46
47
48
    # via
    #   awscli
    #   boto3
    #   s3transfer
49
50
bounded-pool-executor==0.0.3
    # via pqdm
51
buildkite-test-collector==0.1.9
52
    # via -r requirements/test.in
53
54
55
56
57
58
59
60
61
62
63
64
65
66
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
67
    #   jiwer
68
69
    #   nltk
    #   ray
70
    #   typer
71
72
73
74
75
76
77
colorama==0.4.6
    # via
    #   awscli
    #   sacrebleu
    #   tqdm-multiprocess
contourpy==1.3.0
    # via matplotlib
78
79
cramjam==2.9.0
    # via fastparquet
80
81
82
83
cupy-cuda12x==13.3.0
    # via ray
cycler==0.12.1
    # via matplotlib
84
datamodel-code-generator==0.26.3
85
    # via -r requirements/test.in
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
dataproperty==1.0.1
    # via
    #   pytablewriter
    #   tabledata
datasets==3.0.2
    # via
    #   evaluate
    #   lm-eval
decorator==5.1.1
    # via librosa
dill==0.3.8
    # via
    #   datasets
    #   evaluate
    #   lm-eval
    #   multiprocess
dnspython==2.7.0
    # via email-validator
docutils==0.16
    # via awscli
einops==0.8.0
107
    # via
108
    #   -r requirements/test.in
109
110
111
112
113
    #   encodec
    #   vector-quantize-pytorch
    #   vocos
einx==0.3.0
    # via vector-quantize-pytorch
114
115
email-validator==2.2.0
    # via pydantic
116
117
encodec==0.1.1
    # via vocos
118
119
evaluate==0.4.3
    # via lm-eval
120
121
fastparquet==2024.11.0
    # via genai-perf
122
123
fastrlock==0.8.2
    # via cupy-cuda12x
124
125
fastsafetensors==0.1.10
    # via -r requirements/test.in
126
127
128
129
130
131
132
133
134
filelock==3.16.1
    # via
    #   datasets
    #   huggingface-hub
    #   ray
    #   torch
    #   transformers
fonttools==4.54.1
    # via matplotlib
135
136
frozendict==2.4.6
    # via einx
137
138
139
140
141
frozenlist==1.5.0
    # via
    #   aiohttp
    #   aiosignal
    #   ray
142
fsspec==2024.9.0
143
144
145
    # via
    #   datasets
    #   evaluate
146
    #   fastparquet
147
148
    #   huggingface-hub
    #   torch
149
genai-perf==0.0.8
150
    # via -r requirements/test.in
151
152
153
154
genson==1.3.0
    # via datamodel-code-generator
h11==0.14.0
    # via httpcore
155
156
hf-xet==0.1.4
    # via huggingface-hub
157
158
159
160
161
hiredis==3.0.0
    # via tensorizer
httpcore==1.0.6
    # via httpx
httpx==0.27.2
162
    # via -r requirements/test.in
163
huggingface-hub==0.30.1
164
    # via
165
    #   -r requirements/test.in
166
167
168
169
170
171
172
173
    #   accelerate
    #   datasets
    #   evaluate
    #   peft
    #   sentence-transformers
    #   timm
    #   tokenizers
    #   transformers
174
    #   vocos
175
176
humanize==4.11.0
    # via runai-model-streamer
177
178
179
180
181
182
183
184
185
186
187
188
189
idna==3.10
    # via
    #   anyio
    #   email-validator
    #   httpx
    #   requests
    #   yarl
inflect==5.6.2
    # via datamodel-code-generator
iniconfig==2.0.0
    # via pytest
isort==5.13.2
    # via datamodel-code-generator
190
jinja2==3.1.6
191
192
193
    # via
    #   datamodel-code-generator
    #   torch
194
jiwer==3.0.5
195
    # via -r requirements/test.in
196
197
198
199
200
201
202
203
204
205
206
207
jmespath==1.0.1
    # via
    #   boto3
    #   botocore
joblib==1.4.2
    # via
    #   librosa
    #   nltk
    #   scikit-learn
jsonlines==4.0.0
    # via lm-eval
jsonschema==4.23.0
208
209
210
    # via
    #   mistral-common
    #   ray
211
212
jsonschema-specifications==2024.10.1
    # via jsonschema
213
214
kaleido==0.2.1
    # via genai-perf
215
216
217
218
219
220
221
kiwisolver==1.4.7
    # via matplotlib
lazy-loader==0.4
    # via librosa
libnacl==2.1.0
    # via tensorizer
librosa==0.10.2.post1
222
    # via -r requirements/test.in
223
llvmlite==0.44.0
224
    # via numba
225
lm-eval==0.4.8
226
    # via -r requirements/test.in
227
228
lxml==5.3.0
    # via sacrebleu
229
230
markdown-it-py==3.0.0
    # via rich
231
232
233
markupsafe==3.0.2
    # via jinja2
matplotlib==3.9.2
234
    # via -r requirements/test.in
235
236
237
238
239
mbstrdecoder==1.1.3
    # via
    #   dataproperty
    #   pytablewriter
    #   typepy
240
241
mdurl==0.1.2
    # via markdown-it-py
242
mistral-common==1.5.4
243
    # via -r requirements/test.in
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
more-itertools==10.5.0
    # via lm-eval
mpmath==1.3.0
    # via sympy
msgpack==1.1.0
    # via
    #   librosa
    #   ray
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
nltk==3.9.1
    # via rouge-score
266
267
268
269
numba==0.61.0
    # via
    #   -r requirements/test.in
    #   librosa
270
271
272
273
numexpr==2.10.1
    # via lm-eval
numpy==1.26.4
    # via
274
    #   -r requirements/test.in
275
276
277
278
279
    #   accelerate
    #   bitsandbytes
    #   contourpy
    #   cupy-cuda12x
    #   datasets
280
281
    #   einx
    #   encodec
282
    #   evaluate
283
284
    #   fastparquet
    #   genai-perf
285
286
    #   librosa
    #   matplotlib
287
    #   mistral-common
288
289
    #   numba
    #   numexpr
290
    #   opencv-python-headless
291
    #   pandas
292
    #   patsy
293
294
    #   peft
    #   rouge-score
295
    #   runai-model-streamer
296
297
298
299
    #   sacrebleu
    #   scikit-learn
    #   scipy
    #   soxr
300
    #   statsmodels
301
302
303
    #   tensorizer
    #   torchvision
    #   transformers
304
    #   tritonclient
305
    #   vocos
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
nvidia-cublas-cu12==12.4.5.8
    # via
    #   nvidia-cudnn-cu12
    #   nvidia-cusolver-cu12
    #   torch
nvidia-cuda-cupti-cu12==12.4.127
    # via torch
nvidia-cuda-nvrtc-cu12==12.4.127
    # via torch
nvidia-cuda-runtime-cu12==12.4.127
    # via torch
nvidia-cudnn-cu12==9.1.0.70
    # via torch
nvidia-cufft-cu12==11.2.1.3
    # via torch
nvidia-curand-cu12==10.3.5.147
    # via torch
nvidia-cusolver-cu12==11.6.1.9
    # via torch
nvidia-cusparse-cu12==12.3.1.170
    # via
    #   nvidia-cusolver-cu12
    #   torch
Michael Goin's avatar
Michael Goin committed
329
330
nvidia-cusparselt-cu12==0.6.2
    # via torch
331
332
333
334
335
336
337
338
339
nvidia-nccl-cu12==2.21.5
    # via torch
nvidia-nvjitlink-cu12==12.4.127
    # via
    #   nvidia-cusolver-cu12
    #   nvidia-cusparse-cu12
    #   torch
nvidia-nvtx-cu12==12.4.127
    # via torch
340
341
342
343
opencv-python-headless==4.11.0.86
    # via
    #   -r requirements/test.in
    #   mistral-common
344
345
346
347
348
349
350
packaging==24.1
    # via
    #   accelerate
    #   black
    #   datamodel-code-generator
    #   datasets
    #   evaluate
351
    #   fastparquet
352
353
354
355
    #   huggingface-hub
    #   lazy-loader
    #   matplotlib
    #   peft
356
    #   plotly
357
358
359
360
    #   pooch
    #   pytest
    #   pytest-rerunfailures
    #   ray
361
    #   statsmodels
362
363
364
365
366
367
    #   transformers
    #   typepy
pandas==2.2.3
    # via
    #   datasets
    #   evaluate
368
369
370
    #   fastparquet
    #   genai-perf
    #   statsmodels
371
372
373
374
pathspec==0.12.1
    # via black
pathvalidate==3.2.1
    # via pytablewriter
375
376
patsy==1.0.1
    # via statsmodels
377
378
peft==0.13.2
    # via
379
    #   -r requirements/test.in
380
    #   lm-eval
381
pillow==10.4.0
382
    # via
383
    #   genai-perf
384
    #   matplotlib
385
    #   mistral-common
386
387
388
389
390
391
    #   sentence-transformers
    #   torchvision
platformdirs==4.3.6
    # via
    #   black
    #   pooch
392
393
plotly==5.24.1
    # via genai-perf
394
395
396
397
398
399
pluggy==1.5.0
    # via pytest
pooch==1.8.2
    # via librosa
portalocker==2.10.1
    # via sacrebleu
400
pqdm==0.2.0
401
    # via -r requirements/test.in
402
403
404
405
406
407
408
409
410
411
412
413
414
415
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
416
417
418
    # via
    #   datasets
    #   genai-perf
419
420
421
422
423
424
pyasn1==0.6.1
    # via rsa
pybind11==2.13.6
    # via lm-eval
pycparser==2.22
    # via cffi
425
pydantic==2.9.2
426
427
428
    # via
    #   datamodel-code-generator
    #   mistral-common
429
430
pydantic-core==2.23.4
    # via pydantic
431
432
pygments==2.18.0
    # via rich
433
434
435
436
437
438
pyparsing==3.2.0
    # via matplotlib
pytablewriter==1.2.0
    # via lm-eval
pytest==8.3.3
    # via
439
    #   -r requirements/test.in
440
    #   buildkite-test-collector
441
    #   genai-perf
442
443
    #   pytest-asyncio
    #   pytest-forked
444
    #   pytest-mock
445
446
447
    #   pytest-rerunfailures
    #   pytest-shard
pytest-asyncio==0.24.0
448
    # via -r requirements/test.in
449
pytest-forked==1.6.0
450
    # via -r requirements/test.in
451
452
pytest-mock==3.14.0
    # via genai-perf
453
pytest-rerunfailures==14.0
454
    # via -r requirements/test.in
455
pytest-shard==0.1.2
456
    # via -r requirements/test.in
457
458
459
460
461
462
python-dateutil==2.9.0.post0
    # via
    #   botocore
    #   matplotlib
    #   pandas
    #   typepy
463
464
python-rapidjson==1.20
    # via tritonclient
465
466
467
468
469
470
471
472
473
474
pytz==2024.2
    # via
    #   pandas
    #   typepy
pyyaml==6.0.2
    # via
    #   accelerate
    #   awscli
    #   datamodel-code-generator
    #   datasets
475
    #   genai-perf
476
477
478
    #   huggingface-hub
    #   peft
    #   ray
479
    #   responses
480
481
    #   timm
    #   transformers
482
    #   vocos
483
484
rapidfuzz==3.12.1
    # via jiwer
Rui Qiao's avatar
Rui Qiao committed
485
ray==2.43.0
486
    # via -r requirements/test.in
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
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
506
    #   mistral-common
507
508
    #   pooch
    #   ray
509
    #   responses
510
511
    #   tiktoken
    #   transformers
512
513
514
responses==0.25.3
    # via genai-perf
rich==13.9.4
515
516
517
    # via
    #   genai-perf
    #   typer
518
519
rouge-score==0.1.2
    # via lm-eval
520
rpds-py==0.20.1
521
522
523
524
525
    # via
    #   jsonschema
    #   referencing
rsa==4.7.2
    # via awscli
526
runai-model-streamer==0.11.0
527
    # via -r requirements/test.in
528
runai-model-streamer-s3==0.11.0
529
    # via -r requirements/test.in
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
s3transfer==0.10.3
    # via
    #   awscli
    #   boto3
sacrebleu==2.4.3
    # via lm-eval
safetensors==0.4.5
    # via
    #   accelerate
    #   peft
    #   timm
    #   transformers
scikit-learn==1.5.2
    # via
    #   librosa
    #   lm-eval
    #   sentence-transformers
scipy==1.13.1
    # via
    #   librosa
    #   scikit-learn
    #   sentence-transformers
552
    #   statsmodels
553
    #   vocos
554
sentence-transformers==3.2.1
555
    # via -r requirements/test.in
556
557
sentencepiece==0.2.0
    # via mistral-common
558
559
560
561
setuptools==75.8.0
    # via
    #   pytablewriter
    #   torch
562
563
shellingham==1.5.4
    # via typer
564
565
566
567
568
569
570
571
572
573
six==1.16.0
    # via
    #   python-dateutil
    #   rouge-score
sniffio==1.3.1
    # via
    #   anyio
    #   httpx
soundfile==0.12.1
    # via
574
    #   -r requirements/test.in
575
576
577
578
579
    #   librosa
soxr==0.5.0.post1
    # via librosa
sqlitedict==2.1.0
    # via lm-eval
580
581
statsmodels==0.14.4
    # via genai-perf
582
sympy==1.13.1
583
584
585
    # via
    #   einx
    #   torch
586
587
588
589
590
591
592
tabledata==1.3.3
    # via pytablewriter
tabulate==0.9.0
    # via sacrebleu
tcolorpy==0.1.6
    # via pytablewriter
tenacity==9.0.0
593
594
595
    # via
    #   lm-eval
    #   plotly
596
tensorizer==2.9.0
597
    # via -r requirements/test.in
598
599
threadpoolctl==3.5.0
    # via scikit-learn
600
601
602
603
tiktoken==0.7.0
    # via
    #   lm-eval
    #   mistral-common
604
timm==1.0.11
605
    # via -r requirements/test.in
606
tokenizers==0.21.0
607
    # via transformers
Michael Goin's avatar
Michael Goin committed
608
torch==2.6.0
609
    # via
610
    #   -r requirements/test.in
611
612
    #   accelerate
    #   bitsandbytes
613
    #   encodec
614
    #   fastsafetensors
615
616
    #   lm-eval
    #   peft
617
    #   runai-model-streamer
618
619
620
    #   sentence-transformers
    #   tensorizer
    #   timm
621
    #   torchaudio
622
    #   torchvision
623
624
    #   vector-quantize-pytorch
    #   vocos
Michael Goin's avatar
Michael Goin committed
625
torchaudio==2.6.0
626
    # via
627
    #   -r requirements/test.in
628
629
    #   encodec
    #   vocos
Michael Goin's avatar
Michael Goin committed
630
631
632
633
torchvision==0.21.0
    # via
    #   -r requirements/test.in
    #   timm
634
635
636
637
638
639
640
641
tqdm==4.66.6
    # via
    #   datasets
    #   evaluate
    #   huggingface-hub
    #   lm-eval
    #   nltk
    #   peft
642
    #   pqdm
643
644
645
646
647
    #   sentence-transformers
    #   tqdm-multiprocess
    #   transformers
tqdm-multiprocess==0.0.11
    # via lm-eval
648
transformers==4.51.0
649
    # via
650
    #   -r requirements/test.in
651
    #   genai-perf
652
653
654
655
656
    #   lm-eval
    #   peft
    #   sentence-transformers
    #   transformers-stream-generator
transformers-stream-generator==0.0.5
657
    # via -r requirements/test.in
Michael Goin's avatar
Michael Goin committed
658
triton==3.2.0
659
    # via torch
660
661
tritonclient==2.51.0
    # via
662
    #   -r requirements/test.in
663
    #   genai-perf
664
typepy==1.3.2
665
666
667
668
    # via
    #   dataproperty
    #   pytablewriter
    #   tabledata
669
670
typer==0.15.2
    # via fastsafetensors
671
672
673
674
typing-extensions==4.12.2
    # via
    #   huggingface-hub
    #   librosa
675
    #   mistral-common
676
    #   pqdm
677
678
679
    #   pydantic
    #   pydantic-core
    #   torch
680
    #   typer
681
682
tzdata==2024.2
    # via pandas
683
urllib3==2.2.3
684
685
686
    # via
    #   botocore
    #   requests
687
688
    #   responses
    #   tritonclient
689
vector-quantize-pytorch==1.21.2
690
    # via -r requirements/test.in
691
vocos==0.1.0
692
    # via -r requirements/test.in
693
694
695
696
697
698
word2number==1.1
    # via lm-eval
xxhash==3.5.0
    # via
    #   datasets
    #   evaluate
699
yarl==1.17.1
700
701
702
    # via aiohttp
zstandard==0.23.0
    # via lm-eval