diff --git a/Dockerfile b/Dockerfile index c7c65769..a510ea59 100644 --- a/Dockerfile +++ b/Dockerfile @@ -72,7 +72,5 @@ COPY --from=builder /usr/src/target/release/text-generation-launcher /usr/local/ # Final image FROM base -COPY ./tgi-entrypoint.sh /tgi-entrypoint.sh - -ENTRYPOINT ["/tgi-entrypoint.sh"] +ENTRYPOINT ["text-generation-launcher"] CMD ["--json-output"] diff --git a/launcher/src/main.rs b/launcher/src/main.rs index 5b808b62..2bbfc017 100644 --- a/launcher/src/main.rs +++ b/launcher/src/main.rs @@ -1253,6 +1253,16 @@ fn terminate(process_name: &str, mut process: Child, timeout: Duration) -> io::R } fn main() -> Result<(), LauncherError> { + match Command::new("ldconfig").spawn() { + Ok(_) => {} + Err(err) => { + tracing::warn!( + "Unable to refresh ldconfig cache. Skipping (useless in most cases). Details {:?}", + err + ) + } + } + // Pattern match configuration let args: Args = Args::parse(); diff --git a/tgi-entrypoint.sh b/tgi-entrypoint.sh deleted file mode 100755 index ea94dcd9..00000000 --- a/tgi-entrypoint.sh +++ /dev/null @@ -1,5 +0,0 @@ -#!/bin/bash - -ldconfig 2>/dev/null || echo 'unable to refresh ld cache, not a big deal in most cases' - -text-generation-launcher $@