diff --git a/requirements/base.in b/requirements/base.in index 8dfa4cad..7d86eac3 100644 --- a/requirements/base.in +++ b/requirements/base.in @@ -2,7 +2,7 @@ layoutparser python-multipart huggingface-hub -numpy<2 +numpy<3 opencv-python!=4.7.0.68 onnx onnxruntime>=1.17.0 diff --git a/requirements/base.txt b/requirements/base.txt index c5d2d72e..033e5bcb 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -25,6 +25,7 @@ filelock==3.15.4 # huggingface-hub # torch # transformers + # triton flatbuffers==24.3.25 # via onnxruntime fonttools==4.53.0 @@ -35,7 +36,7 @@ fsspec==2024.6.0 # torch huggingface-hub==0.23.4 # via - # -r requirements/base.in + # -r base.in # timm # tokenizers # transformers @@ -52,18 +53,18 @@ jinja2==3.1.4 kiwisolver==1.4.5 # via matplotlib layoutparser==0.3.4 - # via -r requirements/base.in + # via -r base.in markupsafe==2.1.5 # via jinja2 matplotlib==3.9.0 - # via -r requirements/base.in + # via -r base.in mpmath==1.3.0 # via sympy networkx==3.2.1 # via torch -numpy==1.26.4 +numpy==2.0.1 # via - # -r requirements/base.in + # -r base.in # contourpy # layoutparser # matplotlib @@ -74,13 +75,44 @@ numpy==1.26.4 # scipy # torchvision # transformers +nvidia-cublas-cu12==12.1.3.1 + # via + # nvidia-cudnn-cu12 + # nvidia-cusolver-cu12 + # torch +nvidia-cuda-cupti-cu12==12.1.105 + # via torch +nvidia-cuda-nvrtc-cu12==12.1.105 + # via torch +nvidia-cuda-runtime-cu12==12.1.105 + # via torch +nvidia-cudnn-cu12==8.9.2.26 + # via torch +nvidia-cufft-cu12==11.0.2.54 + # via torch +nvidia-curand-cu12==10.3.2.106 + # via torch +nvidia-cusolver-cu12==11.4.5.107 + # via torch +nvidia-cusparse-cu12==12.1.0.106 + # via + # nvidia-cusolver-cu12 + # torch +nvidia-nccl-cu12==2.20.5 + # via torch +nvidia-nvjitlink-cu12==12.5.82 + # via + # nvidia-cusolver-cu12 + # nvidia-cusparse-cu12 +nvidia-nvtx-cu12==12.1.105 + # via torch onnx==1.16.1 - # via -r requirements/base.in + # via -r base.in onnxruntime==1.18.0 - # via -r requirements/base.in + # via -r base.in opencv-python==4.10.0.84 # via - # -r requirements/base.in + # -r base.in # layoutparser packaging==24.1 # via @@ -120,7 +152,7 @@ python-dateutil==2.9.0.post0 # matplotlib # pandas python-multipart==0.0.9 - # via -r requirements/base.in + # via -r base.in pytz==2024.1 # via pandas pyyaml==6.0.1 @@ -130,7 +162,7 @@ pyyaml==6.0.1 # timm # transformers rapidfuzz==3.9.3 - # via -r requirements/base.in + # via -r base.in regex==2024.5.15 # via transformers requests==2.32.3 @@ -150,12 +182,12 @@ sympy==1.12.1 # onnxruntime # torch timm==1.0.7 - # via -r requirements/base.in + # via -r base.in tokenizers==0.19.1 # via transformers torch==2.3.1 # via - # -r requirements/base.in + # -r base.in # timm # torchvision torchvision==0.18.1 @@ -166,7 +198,9 @@ tqdm==4.66.4 # iopath # transformers transformers==4.41.2 - # via -r requirements/base.in + # via -r base.in +triton==2.3.1 + # via torch typing-extensions==4.12.2 # via # huggingface-hub diff --git a/requirements/dev.txt b/requirements/dev.txt index 76db855d..33f6acec 100644 --- a/requirements/dev.txt +++ b/requirements/dev.txt @@ -6,11 +6,9 @@ # anyio==4.4.0 # via - # -c requirements/test.txt + # -c test.txt # httpx # jupyter-server -appnope==0.1.4 - # via ipykernel argon2-cffi==23.1.0 # via jupyter-server argon2-cffi-bindings==21.2.0 @@ -35,23 +33,23 @@ build==1.2.1 # via pip-tools certifi==2024.6.2 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # httpcore # httpx # requests cffi==1.16.0 # via - # -c requirements/base.txt + # -c base.txt # argon2-cffi-bindings charset-normalizer==3.3.2 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # requests click==8.1.7 # via - # -c requirements/test.txt + # -c test.txt # pip-tools comm==0.2.2 # via @@ -59,11 +57,11 @@ comm==0.2.2 # ipywidgets contourpy==1.2.1 # via - # -c requirements/base.txt + # -c base.txt # matplotlib cycler==0.12.1 # via - # -c requirements/base.txt + # -c base.txt # matplotlib debugpy==1.8.1 # via ipykernel @@ -73,7 +71,7 @@ defusedxml==0.7.1 # via nbconvert exceptiongroup==1.2.1 # via - # -c requirements/test.txt + # -c test.txt # anyio # ipython executing==2.0.1 @@ -82,26 +80,26 @@ fastjsonschema==2.20.0 # via nbformat fonttools==4.53.0 # via - # -c requirements/base.txt + # -c base.txt # matplotlib fqdn==1.5.1 # via jsonschema h11==0.14.0 # via - # -c requirements/test.txt + # -c test.txt # httpcore httpcore==1.0.5 # via - # -c requirements/test.txt + # -c test.txt # httpx httpx==0.27.0 # via - # -c requirements/test.txt + # -c test.txt # jupyterlab idna==3.7 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # anyio # httpx # jsonschema @@ -116,7 +114,7 @@ importlib-metadata==8.0.0 # nbconvert importlib-resources==6.4.0 # via - # -c requirements/base.txt + # -c base.txt # matplotlib ipykernel==6.29.4 # via @@ -126,7 +124,7 @@ ipykernel==6.29.4 # qtconsole ipython==8.18.1 # via - # -r requirements/dev.in + # -r dev.in # ipykernel # ipywidgets # jupyter-console @@ -138,7 +136,7 @@ jedi==0.19.1 # via ipython jinja2==3.1.4 # via - # -c requirements/base.txt + # -c base.txt # jupyter-server # jupyterlab # jupyterlab-server @@ -155,7 +153,7 @@ jsonschema[format-nongpl]==4.22.0 jsonschema-specifications==2023.12.1 # via jsonschema jupyter==1.0.0 - # via -r requirements/dev.in + # via -r dev.in jupyter-client==8.6.2 # via # ipykernel @@ -201,17 +199,17 @@ jupyterlab-widgets==3.0.11 # via ipywidgets kiwisolver==1.4.5 # via - # -c requirements/base.txt + # -c base.txt # matplotlib markupsafe==2.1.5 # via - # -c requirements/base.txt + # -c base.txt # jinja2 # nbconvert matplotlib==3.9.0 # via - # -c requirements/base.txt - # -r requirements/dev.in + # -c base.txt + # -r dev.in matplotlib-inline==0.1.7 # via # ipykernel @@ -237,17 +235,17 @@ notebook-shim==0.2.4 # via # jupyterlab # notebook -numpy==1.26.4 +numpy==2.0.1 # via - # -c requirements/base.txt + # -c base.txt # contourpy # matplotlib overrides==7.7.0 # via jupyter-server packaging==24.1 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # build # ipykernel # jupyter-server @@ -265,14 +263,14 @@ pexpect==4.9.0 # via ipython pillow==10.3.0 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # matplotlib pip-tools==7.4.1 - # via -r requirements/dev.in + # via -r dev.in platformdirs==4.2.2 # via - # -c requirements/test.txt + # -c test.txt # jupyter-core prometheus-client==0.20.0 # via jupyter-server @@ -290,7 +288,7 @@ pure-eval==0.2.2 # via stack-data pycparser==2.22 # via - # -c requirements/base.txt + # -c base.txt # cffi pygments==2.18.0 # via @@ -300,7 +298,7 @@ pygments==2.18.0 # qtconsole pyparsing==3.1.2 # via - # -c requirements/base.txt + # -c base.txt # matplotlib pyproject-hooks==1.1.0 # via @@ -308,7 +306,7 @@ pyproject-hooks==1.1.0 # pip-tools python-dateutil==2.9.0.post0 # via - # -c requirements/base.txt + # -c base.txt # arrow # jupyter-client # matplotlib @@ -316,8 +314,8 @@ python-json-logger==2.0.7 # via jupyter-events pyyaml==6.0.1 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # jupyter-events pyzmq==26.0.3 # via @@ -337,8 +335,8 @@ referencing==0.35.1 # jupyter-events requests==2.32.3 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # jupyterlab-server rfc3339-validator==0.1.4 # via @@ -356,14 +354,14 @@ send2trash==1.8.3 # via jupyter-server six==1.16.0 # via - # -c requirements/base.txt + # -c base.txt # asttokens # bleach # python-dateutil # rfc3339-validator sniffio==1.3.1 # via - # -c requirements/test.txt + # -c test.txt # anyio # httpx soupsieve==2.5 @@ -378,7 +376,7 @@ tinycss2==1.3.0 # via nbconvert tomli==2.0.1 # via - # -c requirements/test.txt + # -c test.txt # build # jupyterlab # pip-tools @@ -411,8 +409,8 @@ types-python-dateutil==2.9.0.20240316 # via arrow typing-extensions==4.12.2 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # anyio # async-lru # ipython @@ -420,8 +418,8 @@ uri-template==1.3.0 # via jsonschema urllib3==2.2.2 # via - # -c requirements/base.txt - # -c requirements/test.txt + # -c base.txt + # -c test.txt # requests wcwidth==0.2.13 # via prompt-toolkit @@ -439,7 +437,7 @@ widgetsnbextension==4.0.11 # via ipywidgets zipp==3.19.2 # via - # -c requirements/base.txt + # -c base.txt # importlib-metadata # importlib-resources