restore dockerfile

This commit is contained in:
Wing Lian
2025-04-24 13:11:00 -04:00
parent fb4adbb311
commit 388e950016
2 changed files with 6 additions and 1 deletions

View File

@@ -1,4 +1,4 @@
FROM axolotlai/axolotl-cloud:main-py3.11-cu124-2.6.0
FROM runpod/pytorch:3.10-2.0.0-117
COPY .runpod/requirements.txt /requirements.txt
RUN --mount=type=cache,target=/root/.cache/pip \