tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

restore dockerfile

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
.runpod/Dockerfile
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 COPY .runpod/requirements.txt /requirements.txt
RUN --mount=type=cache,target=/root/.cache/pip \ RUN --mount=type=cache,target=/root/.cache/pip \

5
.runpod/requirements.txt
View File

@@ -5,6 +5,11 @@
# git+https://github.com/runpod/runpod-python.git # git+https://github.com/runpod/runpod-python.git
# To learn more, see https://pip.pypa.io/en/stable/reference/requirements-file-format/ # To learn more, see https://pip.pypa.io/en/stable/reference/requirements-file-format/
runpod~=1.7.0 runpod~=1.7.0
huggingface_hub
typing-extensions typing-extensions
pydantic pydantic
pydantic-settings pydantic-settings
hf-transfer
setuptools
numpy==2.0.0
axolotl[flash-attn,deepspeed]