@@ -25,6 +25,7 @@ filelock==3.15.4
2525 # huggingface-hub
2626 # torch
2727 # transformers
28+ # triton
2829flatbuffers==24.3.25
2930 # via onnxruntime
3031fonttools==4.53.0
@@ -35,7 +36,7 @@ fsspec==2024.6.0
3536 # torch
3637huggingface-hub==0.23.4
3738 # via
38- # -r requirements/ base.in
39+ # -r base.in
3940 # timm
4041 # tokenizers
4142 # transformers
@@ -52,18 +53,18 @@ jinja2==3.1.4
5253kiwisolver==1.4.5
5354 # via matplotlib
5455layoutparser==0.3.4
55- # via -r requirements/ base.in
56+ # via -r base.in
5657markupsafe==2.1.5
5758 # via jinja2
5859matplotlib==3.9.0
59- # via -r requirements/ base.in
60+ # via -r base.in
6061mpmath==1.3.0
6162 # via sympy
6263networkx==3.2.1
6364 # via torch
64- numpy==1.26.4
65+ numpy==2.0.2
6566 # via
66- # -r requirements/ base.in
67+ # -r base.in
6768 # contourpy
6869 # layoutparser
6970 # matplotlib
@@ -74,13 +75,44 @@ numpy==1.26.4
7475 # scipy
7576 # torchvision
7677 # 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==8.9.2.26
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
77109onnx==1.16.1
78- # via -r requirements/ base.in
110+ # via -r base.in
79111onnxruntime==1.18.0
80- # via -r requirements/ base.in
112+ # via -r base.in
81113opencv-python==4.10.0.84
82114 # via
83- # -r requirements/ base.in
115+ # -r base.in
84116 # layoutparser
85117packaging==24.1
86118 # via
@@ -120,7 +152,7 @@ python-dateutil==2.9.0.post0
120152 # matplotlib
121153 # pandas
122154python-multipart==0.0.9
123- # via -r requirements/ base.in
155+ # via -r base.in
124156pytz==2024.1
125157 # via pandas
126158pyyaml==6.0.1
@@ -130,7 +162,7 @@ pyyaml==6.0.1
130162 # timm
131163 # transformers
132164rapidfuzz==3.9.3
133- # via -r requirements/ base.in
165+ # via -r base.in
134166regex==2024.5.15
135167 # via transformers
136168requests==2.32.3
@@ -150,12 +182,12 @@ sympy==1.12.1
150182 # onnxruntime
151183 # torch
152184timm==1.0.7
153- # via -r requirements/ base.in
185+ # via -r base.in
154186tokenizers==0.19.1
155187 # via transformers
156188torch==2.3.1
157189 # via
158- # -r requirements/ base.in
190+ # -r base.in
159191 # timm
160192 # torchvision
161193torchvision==0.18.1
@@ -166,7 +198,9 @@ tqdm==4.66.4
166198 # iopath
167199 # transformers
168200transformers==4.41.2
169- # via -r requirements/base.in
201+ # via -r base.in
202+ triton==2.3.1
203+ # via torch
170204typing-extensions==4.12.2
171205 # via
172206 # huggingface-hub
0 commit comments