.

leloykun · Dec 31, 2024 · aec15c2 · aec15c2
1 parent 92aa0c0
commit aec15c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -27,7 +27,7 @@ WORKDIR /modded-nanogpt
 RUN python -m pip install --upgrade pip && \
     pip install -r requirements.txt
 
-RUN pip install --pre torch==2.6.0.dev20241203+cu126 --index-url https://download.pytorch.org/whl/nightly/cu126 --upgrade
+RUN pip install --pre torch==2.6.0.dev20241231+cu126 --index-url https://download.pytorch.org/whl/nightly/cu126 --upgrade
 
 CMD ["bash"]
 ENTRYPOINT []