update tgi entrypoint

This commit is contained in:
Morgan Funtowicz 2024-07-22 19:13:01 +00:00
parent d0a34a95f2
commit 805e584b92

View File

@ -62,5 +62,5 @@ COPY --from=cuda-builder /usr/local/tensorrt /usr/local/tensorrt
COPY --from=cuda-builder /usr/local/tgi /usr/local/tgi
COPY --from=cuda-builder /usr/src/text-generation-inference/target/release/text-generation-backends-trtllm /usr/local/tgi/bin/text-generation-launcher
ENTRYPOINT ["text-generation-launcher"]
ENTRYPOINT ["./text-generation-launcher"]
CMD ["--executor-worker", "/usr/local/tgi/bin/executorWorker"]