From 4c75f888db73225bed2e73cd91a0bf9fe0e0cbd7 Mon Sep 17 00:00:00 2001 From: Shubham Krishna Date: Wed, 13 Mar 2024 08:49:43 +0100 Subject: [PATCH] Fix TGI Dockerfile (#3) --- text-generation-inference/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/text-generation-inference/Dockerfile b/text-generation-inference/Dockerfile index c8700e6a..fe4d3945 100644 --- a/text-generation-inference/Dockerfile +++ b/text-generation-inference/Dockerfile @@ -15,6 +15,7 @@ ARG CARGO_REGISTRIES_CRATES_IO_PROTOCOL=sparse FROM chef as planner COPY --from=tgi /tgi/Cargo.toml Cargo.toml +COPY --from=tgi /tgi/Cargo.lock Cargo.lock COPY --from=tgi /tgi/rust-toolchain.toml rust-toolchain.toml COPY --from=tgi /tgi/proto proto COPY --from=tgi /tgi/benchmark benchmark @@ -34,6 +35,7 @@ COPY --from=planner /usr/src/recipe.json recipe.json RUN cargo chef cook --release --recipe-path recipe.json COPY --from=tgi /tgi/Cargo.toml Cargo.toml +COPY --from=tgi /tgi/Cargo.lock Cargo.lock COPY --from=tgi /tgi/rust-toolchain.toml rust-toolchain.toml COPY --from=tgi /tgi/proto proto COPY --from=tgi /tgi/benchmark benchmark