diff --git a/.devops/main-cuda.Dockerfile b/.devops/main-cuda.Dockerfile index f48408dbf..083645c77 100644 --- a/.devops/main-cuda.Dockerfile +++ b/.devops/main-cuda.Dockerfile @@ -46,6 +46,7 @@ RUN apt-get update && \ apt-get install -y python3 python3-pip \ && rm -rf /var/lib/apt/lists/* +COPY --from=build /app/.devops/handler.py /handler.py COPY --from=build /app/server /server COPY --from=build /model.gguf model.gguf COPY --from=build /app/models models