From f9ef3933db87920210b45f71be0f0c0dfb933afb Mon Sep 17 00:00:00 2001 From: Shubham Krishna Date: Tue, 12 Mar 2024 19:48:37 +0000 Subject: [PATCH] Fix TGI Dockerfile --- 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