Move model to device using from_pretrained

CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.8.11
+
+* fix: moving the table transformer model to device when loading the model instead of once the model is loaded.
+
 ## 0.8.10
 
 * feat: unpin `numpy` and bump minimum for `onnxruntime` to be compatible with `numpy>=2`

requirements/base.in

@@ -10,6 +10,7 @@ torch
 timm
 # NOTE(alan): Pinned because this is when the most recent module we import appeared
 transformers>=4.25.1
+accelerate
 rapidfuzz
 pandas
 scipy

requirements/base.txt

@@ -1,65 +1,67 @@
 #
-# This file is autogenerated by pip-compile with Python 3.9
+# This file is autogenerated by pip-compile with Python 3.12
 # by the following command:
 #
 #    pip-compile requirements/base.in
 #
-certifi==2025.1.31
+accelerate==1.7.0
+    # via -r requirements/base.in
+certifi==2025.4.26
     # via requests
 cffi==1.17.1
     # via cryptography
-charset-normalizer==3.4.1
+charset-normalizer==3.4.2
     # via
     #   pdfminer-six
     #   requests
 coloredlogs==15.0.1
     # via onnxruntime
-contourpy==1.3.0
+contourpy==1.3.2
     # via matplotlib
-cryptography==44.0.2
+cryptography==44.0.3
     # via pdfminer-six
 cycler==0.12.1
     # via matplotlib
-filelock==3.17.0
+filelock==3.18.0
     # via
     #   huggingface-hub
     #   torch
     #   transformers
 flatbuffers==25.2.10
     # via onnxruntime
-fonttools==4.56.0
+fonttools==4.58.0
     # via matplotlib
-fsspec==2025.3.0
+fsspec==2025.3.2
     # via
     #   huggingface-hub
     #   torch
-huggingface-hub==0.29.3
+huggingface-hub==0.31.2
     # via
     #   -r requirements/base.in
+    #   accelerate
     #   timm
     #   tokenizers
     #   transformers
 humanfriendly==10.0
     # via coloredlogs
 idna==3.10
     # via requests
-importlib-resources==6.5.2
-    # via matplotlib
 jinja2==3.1.6
     # via torch
-kiwisolver==1.4.7
+kiwisolver==1.4.8
     # via matplotlib
 markupsafe==3.0.2
     # via jinja2
-matplotlib==3.9.4
+matplotlib==3.10.3
     # via -r requirements/base.in
 mpmath==1.3.0
     # via sympy
-networkx==3.2.1
+networkx==3.4.2
     # via torch
-numpy==2.0.2
+numpy==2.2.5
     # via
     #   -r requirements/base.in
+    #   accelerate
     #   contourpy
     #   matplotlib
     #   onnx
@@ -69,33 +71,36 @@ numpy==2.0.2
     #   scipy
     #   torchvision
     #   transformers
-onnx==1.17.0
+onnx==1.18.0
     # via -r requirements/base.in
-onnxruntime==1.19.2
+onnxruntime==1.22.0
     # via -r requirements/base.in
 opencv-python==4.11.0.86
     # via -r requirements/base.in
-packaging==24.2
+packaging==25.0
     # via
+    #   accelerate
     #   huggingface-hub
     #   matplotlib
     #   onnxruntime
     #   transformers
 pandas==2.2.3
     # via -r requirements/base.in
-pdfminer-six==20240706
+pdfminer-six==20250506
     # via -r requirements/base.in
-pillow==11.1.0
+pillow==11.2.1
     # via
     #   matplotlib
     #   torchvision
-protobuf==6.30.0
+protobuf==6.31.0
     # via
     #   onnx
     #   onnxruntime
+psutil==7.0.0
+    # via accelerate
 pycparser==2.22
     # via cffi
-pyparsing==3.2.1
+pyparsing==3.2.3
     # via matplotlib
 pypdfium2==4.30.1
     # via -r requirements/base.in
@@ -105,14 +110,15 @@ python-dateutil==2.9.0.post0
     #   pandas
 python-multipart==0.0.20
     # via -r requirements/base.in
-pytz==2025.1
+pytz==2025.2
     # via pandas
 pyyaml==6.0.2
     # via
+    #   accelerate
     #   huggingface-hub
     #   timm
     #   transformers
-rapidfuzz==3.12.2
+rapidfuzz==3.13.0
     # via -r requirements/base.in
 regex==2024.11.6
     # via transformers
@@ -122,40 +128,44 @@ requests==2.32.3
     #   transformers
 safetensors==0.5.3
     # via
+    #   accelerate
     #   timm
     #   transformers
-scipy==1.13.1
+scipy==1.15.3
     # via -r requirements/base.in
 six==1.17.0
     # via python-dateutil
-sympy==1.13.1
+sympy==1.14.0
     # via
     #   onnxruntime
     #   torch
 timm==1.0.15
     # via -r requirements/base.in
 tokenizers==0.21.1
     # via transformers
-torch==2.6.0
+torch==2.7.0
     # via
     #   -r requirements/base.in
+    #   accelerate
     #   timm
     #   torchvision
-torchvision==0.21.0
+torchvision==0.22.0
     # via timm
 tqdm==4.67.1
     # via
     #   huggingface-hub
     #   transformers
-transformers==4.49.0
+transformers==4.51.3
     # via -r requirements/base.in
-typing-extensions==4.12.2
+typing-extensions==4.13.2
     # via
     #   huggingface-hub
+    #   onnx
     #   torch
-tzdata==2025.1
+tzdata==2025.2
     # via pandas
-urllib3==2.3.0
+urllib3==2.4.0
     # via requests
-zipp==3.21.0
-    # via importlib-resources
+
+# The following packages are considered to be unsafe in a requirements file:
+# setuptools