Update Dockerfile

This commit is contained in:
Justin Merrell
2023-12-08 20:42:15 -05:00
parent deb6f67a25
commit bcf37e7bd2
+1 -1
View File
@@ -28,7 +28,7 @@ ADD src .
ARG MODEL_NAME=""
ARG MODEL_BASE_PATH=""
ARG HUGGING_FACE_HUB_TOKEN=""
ARG HF_TOKEN=""
ENV HF_TOKEN=$HF_TOKEN
# Conditionally run download_model.py