@@ -25,6 +25,7 @@ filelock==3.16.0
25
25
# huggingface-hub
26
26
# torch
27
27
# transformers
28
+ # triton
28
29
flatbuffers==24.3.25
29
30
# via onnxruntime
30
31
fonttools==4.53.1
@@ -35,7 +36,7 @@ fsspec==2024.9.0
35
36
# torch
36
37
huggingface-hub==0.24.7
37
38
# via
38
- # -r requirements/ base.in
39
+ # -r base.in
39
40
# timm
40
41
# tokenizers
41
42
# transformers
@@ -52,18 +53,18 @@ jinja2==3.1.4
52
53
kiwisolver==1.4.7
53
54
# via matplotlib
54
55
layoutparser==0.3.4
55
- # via -r requirements/ base.in
56
+ # via -r base.in
56
57
markupsafe==2.1.5
57
58
# via jinja2
58
59
matplotlib==3.9.2
59
- # via -r requirements/ base.in
60
+ # via -r base.in
60
61
mpmath==1.3.0
61
62
# via sympy
62
63
networkx==3.2.1
63
64
# via torch
64
- numpy==1.26.4
65
+ numpy==2.0.2
65
66
# via
66
- # -r requirements/ base.in
67
+ # -r base.in
67
68
# contourpy
68
69
# layoutparser
69
70
# matplotlib
@@ -74,13 +75,44 @@ numpy==1.26.4
74
75
# scipy
75
76
# torchvision
76
77
# transformers
78
+ nvidia-cublas-cu12==12.1.3.1
79
+ # via
80
+ # nvidia-cudnn-cu12
81
+ # nvidia-cusolver-cu12
82
+ # torch
83
+ nvidia-cuda-cupti-cu12==12.1.105
84
+ # via torch
85
+ nvidia-cuda-nvrtc-cu12==12.1.105
86
+ # via torch
87
+ nvidia-cuda-runtime-cu12==12.1.105
88
+ # via torch
89
+ nvidia-cudnn-cu12==9.1.0.70
90
+ # via torch
91
+ nvidia-cufft-cu12==11.0.2.54
92
+ # via torch
93
+ nvidia-curand-cu12==10.3.2.106
94
+ # via torch
95
+ nvidia-cusolver-cu12==11.4.5.107
96
+ # via torch
97
+ nvidia-cusparse-cu12==12.1.0.106
98
+ # via
99
+ # nvidia-cusolver-cu12
100
+ # torch
101
+ nvidia-nccl-cu12==2.20.5
102
+ # via torch
103
+ nvidia-nvjitlink-cu12==12.6.68
104
+ # via
105
+ # nvidia-cusolver-cu12
106
+ # nvidia-cusparse-cu12
107
+ nvidia-nvtx-cu12==12.1.105
108
+ # via torch
77
109
onnx==1.16.2
78
- # via -r requirements/ base.in
110
+ # via -r base.in
79
111
onnxruntime==1.19.2
80
- # via -r requirements/ base.in
112
+ # via -r base.in
81
113
opencv-python==4.10.0.84
82
114
# via
83
- # -r requirements/ base.in
115
+ # -r base.in
84
116
# layoutparser
85
117
packaging==24.1
86
118
# via
@@ -120,7 +152,7 @@ python-dateutil==2.9.0.post0
120
152
# matplotlib
121
153
# pandas
122
154
python-multipart==0.0.9
123
- # via -r requirements/ base.in
155
+ # via -r base.in
124
156
pytz==2024.2
125
157
# via pandas
126
158
pyyaml==6.0.2
@@ -130,7 +162,7 @@ pyyaml==6.0.2
130
162
# timm
131
163
# transformers
132
164
rapidfuzz==3.9.7
133
- # via -r requirements/ base.in
165
+ # via -r base.in
134
166
regex==2024.9.11
135
167
# via transformers
136
168
requests==2.32.3
@@ -150,12 +182,12 @@ sympy==1.13.2
150
182
# onnxruntime
151
183
# torch
152
184
timm==1.0.9
153
- # via -r requirements/ base.in
185
+ # via -r base.in
154
186
tokenizers==0.19.1
155
187
# via transformers
156
188
torch==2.4.1
157
189
# via
158
- # -r requirements/ base.in
190
+ # -r base.in
159
191
# timm
160
192
# torchvision
161
193
torchvision==0.19.1
@@ -166,7 +198,9 @@ tqdm==4.66.5
166
198
# iopath
167
199
# transformers
168
200
transformers==4.44.2
169
- # via -r requirements/base.in
201
+ # via -r base.in
202
+ triton==3.0.0
203
+ # via torch
170
204
typing-extensions==4.12.2
171
205
# via
172
206
# huggingface-hub
0 commit comments