Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
jianlins authored Feb 7, 2025
1 parent b09a0d1 commit 8f57048
Showing 1 changed file with 1 addition and 8 deletions.
9 changes: 1 addition & 8 deletions cosmos_openwebui/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,12 +1,5 @@
FROM ghcr.io/open-webui/open-webui:main
ARG UID=0
ARG GID=0
FROM ghcr.io/open-webui/open-webui:main AS build

ENV WHISPER_MODEL_DIR="/app/backend/data/cache/whisper/models" \
RAG_EMBEDDING_MODEL="sentence-transformers/all-MiniLM-L6-v2" \
SENTENCE_TRANSFORMERS_HOME="/app/backend/data/cache/embedding/models" \
TIKTOKEN_ENCODING_NAME="cl100k_base" \
TIKTOKEN_CACHE_DIR="/app/backend/data/cache/tiktoken"

RUN python -c "import os; from sentence_transformers import SentenceTransformer; SentenceTransformer(os.environ['RAG_EMBEDDING_MODEL'], device='cpu')" && \
python -c "import os; from faster_whisper import WhisperModel; WhisperModel(os.environ['WHISPER_MODEL'], device='cpu', compute_type='int8', download_root=os.environ['WHISPER_MODEL_DIR'])"; \
Expand Down

0 comments on commit 8f57048

Please sign in to comment.