tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

no internet connection for whisper if you use docker

Browse source
This commit is contained in:
Jannik Streidl 2024-02-13 15:11:53 +01:00
parent a8a0ebf3aa
commit 8ab0338e71
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Dockerfile
View file

@ -30,6 +30,10 @@ ENV WEBUI_SECRET_KEY ""
ENV SCARF_NO_ANALYTICS true
ENV DO_NOT_TRACK true
#Whisper TTS Settings
ENV WHISPER_DIR="/app/backend/data/cache/whisper/models"
ENV WHISPER_MODEL="base"
WORKDIR /app/backend
# install python dependencies
@ -45,6 +49,8 @@ RUN apt-get update \
&& rm -rf /var/lib/apt/lists/*
# RUN python -c "from sentence_transformers import SentenceTransformer; model = SentenceTransformer('all-MiniLM-L6-v2')"
RUN python -c "import os; from faster_whisper import WhisperModel; WhisperModel(os.environ['WHISPER_MODEL'], device='cpu', compute_type='int8', download_root=os.environ['WHISPER_DIR'])"
# copy embedding weight from build
RUN mkdir -p /root/.cache/chroma/onnx_models/all-MiniLM-L6-v2