Fixed MODEL_REVISION environment variable

2024-05-02 13:45:43 +02:00
parent 0a5b5bc095
commit f06a64d5b9
1 changed files with 1 additions and 1 deletions
@@ -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 \