Files changed (1) hide show
  1. Dockerfile +3 -20
Dockerfile CHANGED
@@ -1,21 +1,4 @@
1
- FROM docker
 
2
 
3
- USER root
4
-
5
- RUN apk add --no-cache git bash
6
-
7
- ARG REPO_URL=https://github.com/nroggendorff/train-llama.git
8
-
9
- WORKDIR /workspace
10
- RUN git clone $REPO_URL .
11
-
12
- RUN docker build Build \
13
- -f Dockerfile \
14
- --build-arg HF_TOKEN=${cat /run/secrets/HF_TOKEN} \
15
- -t nroggendorff/train-llama:latest
16
-
17
- RUN --mount=type=secret,id=DOCKER_PASSWORD,mode=0444,required=true \
18
- cat /run/secrets/DOCKER_PASSWORD | \
19
- docker login --username "nroggendorff" --password-stdin
20
-
21
- CMD ["docker push", "nroggendorff/train-llama:latest"]
 
1
+ FROM nroggendorff/train-llama:latest
2
+ RUN python -u prep.py
3
 
4
+ CMD ["python", "train.py"]