Spaces:
Paused
Paused
Commit
•
844a3fd
1
Parent(s):
b81453a
Merge when ready EhW6[mT~
Browse files- Dockerfile +6 -1
Dockerfile
CHANGED
|
@@ -1,6 +1,11 @@
|
|
| 1 |
FROM nroggendorff/train-llama:latest
|
| 2 |
|
| 3 |
-
USER
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
RUN python -u prep.py
|
| 5 |
|
| 6 |
CMD ["python", "train.py"]
|
|
|
|
| 1 |
FROM nroggendorff/train-llama:latest
|
| 2 |
|
| 3 |
+
USER root
|
| 4 |
+
|
| 5 |
+
ARG CONFIG="config.json"
|
| 6 |
+
|
| 7 |
+
RUN tmp=$(mktemp) \
|
| 8 |
+
jq '.init = "abcde"' "${CONFIG}" > "$tmp" && mv "$tmp" "${CONFIG}"
|
| 9 |
RUN python -u prep.py
|
| 10 |
|
| 11 |
CMD ["python", "train.py"]
|