harrim-nv commited on
Commit
59f0fba
·
verified ·
1 Parent(s): 6cdf1cf

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +30 -5
requirements.txt CHANGED
@@ -1,6 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  accelerate
2
- diffusers
3
- invisible_watermark
4
- torch
5
- transformers
6
- xformers
 
1
+ cmake
2
+ fastapi
3
+ uvicorn
4
+ transformers>=4.36.0
5
+ datasets
6
+ pydantic
7
+ math_verify
8
+ pybind11[global]
9
+ vllm>=0.3.0
10
+ torch==2.6.0
11
+ torchvision==0.21.0
12
+ tensordict==0.7.2
13
+ toml
14
+ tomli===2.2.1
15
+ StrEnum
16
+ redis
17
+ msgpack
18
+ qwen-vl-utils==0.0.11
19
+ flash-attn==2.3.6
20
+ setuptools==79.0.1
21
+ pynvml
22
+ torchao
23
+ gradio>=4.0.0
24
+ ninja
25
+ packaging
26
  accelerate
27
+ safetensors
28
+ numpy<2.0.0
29
+ pandas>=2.0.0,<2.2.3dev0
30
+ scikit-learn<2.0.0
31
+ pyarrow>=16.1.0,<16.2.0a0