Spaces:
Starting
on
L40S
Starting
on
L40S
Merge when ready EhW6[mT~ (#23)
Browse files- Merge when ready EhW6[mT~ (844a3fd104db6eb0017c67a05e46c6b7f8a326f3)
- 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"]
|