owen/worker-vllm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update Dockerfile

Browse Source
This commit is contained in:
Justin Merrell
2023-12-08 20:28:44 -05:00
parent 35a74475f2
commit 11e6495304
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+1
View File
@@ -29,6 +29,7 @@ ADD src .
ARG MODEL_NAME=""
ARG MODEL_BASE_PATH=""
ARG HUGGING_FACE_HUB_TOKEN=""
ENV HUGGING_FACE_HUB_TOKEN=$HUGGING_FACE_HUB_TOKEN
# Conditionally run download_model.py
RUN if [ -n "$MODEL_NAME" ] && [ -n "$MODEL_BASE_PATH" ]; then \