Update Dockerfile

This commit is contained in:
Justin Merrell
2023-11-29 22:32:13 -05:00
parent c455508931
commit 4131b7a16d
-8
View File
@@ -16,14 +16,6 @@ RUN --mount=type=cache,target=/root/.cache/pip \
ARG MODEL_NAME=""
ENV MODEL_NAME=$MODEL_NAME
# Tokenizer is a path ot a huggingface repo
ARG TOKENIZER=
ENV TOKENIZER=$TOKENIZER
# A string awq or squeezellm
ARG QUANTIZATION=
ENV QUANTIZATION=$QUANTIZATION
# Add source files
ADD src .