From b09f6d36aafefdb14ec2352839c146a24c7972c8 Mon Sep 17 00:00:00 2001 From: Jeffrey Gordon Date: Sat, 30 Nov 2024 16:23:40 -0500 Subject: [PATCH] update whisper --- apps/whisper/Dockerfile | 5 +---- apps/whisper/entrypoint.sh | 8 ++------ 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/apps/whisper/Dockerfile b/apps/whisper/Dockerfile index 5ae7b49..22aec7e 100644 --- a/apps/whisper/Dockerfile +++ b/apps/whisper/Dockerfile @@ -12,9 +12,6 @@ apt-get update \ python3-dev \ python3-pip \ \ -# && pip3 uninstall torch \ -# && pip3 cache purge \ -# && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 \ && pip3 install --no-cache-dir -U \ setuptools \ wheel \ @@ -31,4 +28,4 @@ COPY ./apps/whisper/entrypoint.sh /entrypoint.sh ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu EXPOSE 10300 -ENTRYPOINT ["/entrypoint.sh"] +ENTRYPOINT ["bash", "/entrypoint.sh"] diff --git a/apps/whisper/entrypoint.sh b/apps/whisper/entrypoint.sh index 794a01e..fac9b74 100755 --- a/apps/whisper/entrypoint.sh +++ b/apps/whisper/entrypoint.sh @@ -1,10 +1,6 @@ #!/usr/bin/env bash -#shellcheck disable=SC2086 - -exec \ - python3 -m wyoming_faster_whisper \ +python3 -m wyoming_faster_whisper \ --uri 'tcp://0.0.0.0:10300' \ --data-dir /data \ + --device cuda \ --download-dir /data "$@" - --download-dir /data "$@" \ - --device cuda