diff --git a/Dockerfile b/Dockerfile index 53c2486..960b1c9 100644 --- a/Dockerfile +++ b/Dockerfile @@ -34,6 +34,7 @@ RUN if [ "${USE_ROCM}" = "1" ]; then mv /app/requirements-rocm.txt /app/requirem RUN --mount=type=cache,target=/root/.cache/pip pip install -r requirements.txt COPY *.py *.sh *.default.yaml README.md LICENSE /app/ +COPY scripts/ /app/scripts/ ARG PRELOAD_MODEL ENV PRELOAD_MODEL=${PRELOAD_MODEL}