diff --git a/poetry.lock b/poetry.lock
index 7be21e9653..2d48511f65 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:88495068650500a25401ed83cf3f2c4eb2216a1c3f8c2dcda0fa33bfd801b85f
-size 560742
+oid sha256:21c5c5a18e0af76c6e96cdd9dc8aab02c62ea7741831081f1d73ee66572f2376
+size 563078
diff --git a/pyproject.toml b/pyproject.toml
index c890b78875..d26fc7ff1d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -147,6 +147,7 @@ mpld3 = "^0.5.8"
 msgpack-python = "^0.5.6"
 networkx = "~2.3"
 nvidia-ml-py3 = "^7.352.0"
+onnx2torch = "^1.5.4"
 onnxoptimizer = "^0.3.1"
 opencv-python-headless = { url = "https://github.com/commaai/opencv-python-builder/releases/download/4.5.5.64%2Bcu113/opencv_python_headless-4.5.5.64-cp38-cp38-manylinux_2_31_x86_64.whl" }
 osmium = "^3.3.0"