add apex to dependencies (#29139)

* apex back * update lock --------- Co-authored-by: ZwX1616 <zwx1616@gmail.com> old-commit-hash: 2a027b9179
2 years ago · 69b4fc489b
parent 84108b1872
commit 69b4fc489b
2 changed files with 3 additions and 2 deletions
--- a/poetry.lock
+++ b/poetry.lock
@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9d7c148fcbb968b9a1fc8a5bbba6baaf14a8e80decc5da3c9dc3d88c6758c8b7
-size 778425
+oid sha256:740934eaa6ba344c6589cdcf68f0e791fac2298992355d4005410e43fc31744f
+size 778659
--- a/pyproject.toml
+++ b/pyproject.toml
@ -175,6 +175,7 @@ tritonclient = {version = "2.28.0", extras = ["http"]}
 transformers = "*"
 timm = "==0.9.2"
 PyNvCodec = { git = "https://github.com/NVIDIA/VideoProcessingFramework.git", rev = "3347e55" }
+apex = { url = "https://github.com/commaai/apex/releases/download/pytorch2.0.1%2Bcu11.8/apex-0.1-cp311-cp311-linux_x86_64.whl" }


 [build-system]