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 c9a0bad commit a23c40f
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions cosmos_openwebui/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,7 @@ FROM ghcr.io/open-webui/open-webui:main



RUN pip3 install uv && \
# If you use CUDA the whisper and embedding model will be downloaded on first use
python -c "import os; from sentence_transformers import SentenceTransformer; SentenceTransformer(os.environ['RAG_EMBEDDING_MODEL'], device='cpu')" && \
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'])"; \
python -c "import os; import tiktoken; tiktoken.get_encoding(os.environ['TIKTOKEN_ENCODING_NAME'])"; \
chown -R $UID:$GID /app/backend/data/
Expand Down

0 comments on commit a23c40f

Please sign in to comment.