bump versions of deps (#1621)

* bump versions of deps * bump transformers too * fix xformers deps and include s3fs install
2024-05-15 13:27:44 -04:00
parent 4fde300e5f
commit 039e2a0370
3 changed files with 19 additions and 16 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -11,7 +11,7 @@ ARG PYTORCH_VERSION="2.1.2"
 ENV PYTORCH_VERSION=$PYTORCH_VERSION

 RUN apt-get update && \
-    apt-get install -y --allow-change-held-packages vim curl nano libnccl2 libnccl-dev
+    apt-get install -y --allow-change-held-packages vim curl nano libnccl2 libnccl-dev s3fs

 WORKDIR /workspace