df

viraweb123 · Aug 14, 2024 · f3f4f28 · f3f4f28
1 parent 7c12a3b
commit f3f4f28
Showing 1 changed file with 0 additions and 18 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,21 +1,3 @@
 FROM 139.99.239.192:5000/nvcr.io/nvidia/pytorch:23.12-py3
 
-ENV HF_HOME=/train/.cache/huggingface
-
 RUN pip3 install --no-cache-dir transformers[torch]
-
-WORKDIR /train/code/
-
-
-COPY ./train/code/ /train/code/
-COPY ./train/input/ /train/input/
-COPY ./train/log/ /train/log/
-
-
-RUN mkdir -p /train/.cache/huggingface
-
-
-RUN mkdir -p /train/output && chmod -R 777 /train/output /train/log
-
-
-ENTRYPOINT ["python3", "main.py"] > /train/log/docker.log 2>&1