Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: pin torch version #409

Closed
wants to merge 1 commit into from
Closed

feat: pin torch version #409

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 4 additions & 2 deletions requirements/base.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,14 @@ opencv-python!=4.7.0.68
onnx
onnxruntime>=1.17.0
matplotlib
torch
# NOTE (yao): torch is a core dependency and its version changes may change model results so we pin
# it to specific verions and only bump via intentional PRs
torch==2.5.1
timm
# NOTE(alan): Pinned because this is when the most recent module we import appeared
transformers>=4.25.1
rapidfuzz
pandas
scipy
pypdfium2
pdfminer-six==20240706
pdfminer-six==20240706
18 changes: 9 additions & 9 deletions requirements/base.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#
# pip-compile requirements/base.in
#
certifi==2024.12.14
certifi==2025.1.31
# via requests
cffi==1.17.1
# via cryptography
Expand All @@ -20,20 +20,20 @@ cryptography==44.0.0
# via pdfminer-six
cycler==0.12.1
# via matplotlib
filelock==3.16.1
filelock==3.17.0
# via
# huggingface-hub
# torch
# transformers
flatbuffers==24.12.23
flatbuffers==25.1.24
# via onnxruntime
fonttools==4.55.4
fonttools==4.55.8
# via matplotlib
fsspec==2024.12.0
# via
# huggingface-hub
# torch
huggingface-hub==0.27.1
huggingface-hub==0.28.1
# via
# -r requirements/base.in
# timm
Expand Down Expand Up @@ -105,14 +105,14 @@ python-dateutil==2.9.0.post0
# pandas
python-multipart==0.0.20
# via -r requirements/base.in
pytz==2024.2
pytz==2025.1
# via pandas
pyyaml==6.0.2
# via
# huggingface-hub
# timm
# transformers
rapidfuzz==3.11.0
rapidfuzz==3.12.1
# via -r requirements/base.in
regex==2024.11.6
# via transformers
Expand Down Expand Up @@ -147,13 +147,13 @@ tqdm==4.67.1
# via
# huggingface-hub
# transformers
transformers==4.48.1
transformers==4.48.2
# via -r requirements/base.in
typing-extensions==4.12.2
# via
# huggingface-hub
# torch
tzdata==2024.2
tzdata==2025.1
# via pandas
urllib3==2.3.0
# via requests
Expand Down
1 change: 1 addition & 0 deletions requirements/constraints.in
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
tritonserver[gprc]
18 changes: 9 additions & 9 deletions requirements/dev.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ asttokens==3.0.0
# via stack-data
async-lru==2.0.4
# via jupyterlab
attrs==24.3.0
attrs==25.1.0
# via
# jsonschema
# referencing
Expand All @@ -33,7 +33,7 @@ bleach[css]==6.2.0
# via nbconvert
build==1.2.2.post1
# via pip-tools
certifi==2024.12.14
certifi==2025.1.31
# via
# -c requirements/base.txt
# -c requirements/test.txt
Expand Down Expand Up @@ -76,11 +76,11 @@ exceptiongroup==1.2.2
# -c requirements/test.txt
# anyio
# ipython
executing==2.1.0
executing==2.2.0
# via stack-data
fastjsonschema==2.21.1
# via nbformat
fonttools==4.55.4
fonttools==4.55.8
# via
# -c requirements/base.txt
# matplotlib
Expand Down Expand Up @@ -186,7 +186,7 @@ jupyter-server==2.15.0
# notebook-shim
jupyter-server-terminals==0.5.3
# via jupyter-server
jupyterlab==4.3.4
jupyterlab==4.3.5
# via
# jupyter
# notebook
Expand Down Expand Up @@ -215,11 +215,11 @@ matplotlib-inline==0.1.7
# via
# ipykernel
# ipython
mistune==3.1.0
mistune==3.1.1
# via nbconvert
nbclient==0.10.2
# via nbconvert
nbconvert==7.16.5
nbconvert==7.16.6
# via
# jupyter
# jupyter-server
Expand Down Expand Up @@ -315,13 +315,13 @@ pyyaml==6.0.2
# -c requirements/base.txt
# -c requirements/test.txt
# jupyter-events
pyzmq==26.2.0
pyzmq==26.2.1
# via
# ipykernel
# jupyter-client
# jupyter-console
# jupyter-server
referencing==0.36.1
referencing==0.36.2
# via
# jsonschema
# jsonschema-specifications
Expand Down
10 changes: 5 additions & 5 deletions requirements/test.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
#
anyio==4.8.0
# via httpx
black==24.10.0
black==25.1.0
# via -r requirements/test.in
certifi==2024.12.14
certifi==2025.1.31
# via
# -c requirements/base.txt
# httpcore
Expand All @@ -30,7 +30,7 @@ exceptiongroup==1.2.2
# via
# anyio
# pytest
filelock==3.16.1
filelock==3.17.0
# via
# -c requirements/base.txt
# huggingface-hub
Expand All @@ -50,7 +50,7 @@ httpcore==1.0.7
# via httpx
httpx==0.28.1
# via -r requirements/test.in
huggingface-hub==0.27.1
huggingface-hub==0.28.1
# via
# -c requirements/base.txt
# -r requirements/test.in
Expand Down Expand Up @@ -110,7 +110,7 @@ requests==2.32.3
# via
# -c requirements/base.txt
# huggingface-hub
ruff==0.9.2
ruff==0.9.4
# via -r requirements/test.in
sniffio==1.3.1
# via anyio
Expand Down