Merge when ready

#13
by nroggendorff - opened
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 --mount=type=secret,id=HF_TOKEN,mode=0444 \
13
- docker build -f Build/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 \
18
- echo $(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"]