deprecate py 3.9 support, set min pytorch version (#1343) [skip ci]

2024-02-28 12:58:05 -05:00
parent 0f985e12fe
commit 6d4bbb877f
5 changed files with 7 additions and 7 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -6,7 +6,7 @@ ARG AXOLOTL_EXTRAS=""
 ARG AXOLOTL_ARGS=""
 ARG CUDA="118"
 ENV BNB_CUDA_VERSION=$CUDA
-ARG PYTORCH_VERSION="2.0.1"
+ARG PYTORCH_VERSION="2.1.2"

 ENV PYTORCH_VERSION=$PYTORCH_VERSION