nroggendorff commited on
Commit
a68ca7c
1 Parent(s): 6ddad29

Merge when ready

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -22
Dockerfile CHANGED
@@ -1,23 +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 systemctl start docker
13
-
14
- RUN --mount=type=secret,id=HF_TOKEN,mode=0444 \
15
- docker build -f Build/Dockerfile \
16
- --build-arg HF_TOKEN=$(cat /run/secrets/HF_TOKEN) \
17
- -t nroggendorff/train-llama:latest .
18
-
19
- RUN --mount=type=secret,id=DOCKER_PASSWORD,mode=0444 \
20
- echo $(cat /run/secrets/DOCKER_PASSWORD) | \
21
- docker login --username "nroggendorff" --password-stdin
22
-
23
- 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"]