Upload folder using huggingface_hub

Dockerfile

@@ -1,4 +1,21 @@
-FROM nroggendorff/train-llama:latest
-RUN python -u prep.py
+FROM docker
 
-CMD ["python", "train.py"]
+USER root
+
+RUN apk add --no-cache git bash
+
+ARG REPO_URL=https://github.com/nroggendorff/train-llama.git
+
+WORKDIR /workspace
+RUN git clone $REPO_URL .
+
+RUN docker build Build \
+-f Dockerfile \
+--build-arg HF_TOKEN=${cat /run/secrets/HF_TOKEN} \
+-t nroggendorff/train-llama:latest
+
+RUN --mount=type=secret,id=DOCKER_PASSWORD,mode=0444,required=true \
+cat /run/secrets/DOCKER_PASSWORD | \
+docker login --username "nroggendorff" --password-stdin
+
+CMD ["docker push", "nroggendorff/train-llama:latest"]