Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
jianlins authored Feb 11, 2025
1 parent ad002fc commit 90984ea
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion cosmos_openwebui/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
FROM ghcr.io/open-webui/open-webui:main as base
FROM ghcr.io/open-webui/open-webui:main as build

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'])"; \
Expand Down

0 comments on commit 90984ea

Please sign in to comment.