diff --git a/Dockerfile b/Dockerfile index 3f1bf2f..45f2d66 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,7 +19,7 @@ ARG MODEL_REVISION="" ARG TOKENIZER_REVISION="" ENV MODEL_NAME=$MODEL_NAME \ - MODEL_REVISION=$REVISION \ + MODEL_REVISION=$MODEL_REVISION \ TOKENIZER_NAME=$TOKENIZER_NAME \ TOKENIZER_REVISION=$TOKENIZER_REVISION \ BASE_PATH=$BASE_PATH \