diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm index 76d9091c..89881a7d 100644 --- a/Dockerfile_trtllm +++ b/Dockerfile_trtllm @@ -150,10 +150,4 @@ COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt COPY --from=tgi-builder /usr/local/tgi /usr/local/tgi # Basically we copy from target/debug instead of target/release -COPY --from=tgi-builder /usr/src/text-generation-inference/target/debug/text-generation-backends-trtllm /usr/local/tgi/bin/text-generation-launcher - -FROM ci-runtime - -LABEL co.huggingface.vendor="Hugging Face Inc." -LABEL org.opencontainers.image.authors="hardware@hf.co" -LABEL org.opencontainers.title="Text-Generation-Inference TensorRT-LLM Backend CI/CD" \ No newline at end of file +COPY --from=tgi-builder /usr/src/text-generation-inference/target/debug/text-generation-backends-trtllm /usr/local/tgi/bin/text-generation-launcher \ No newline at end of file