From 8f570480db454b31cd926f0f5fd2df95bda691d5 Mon Sep 17 00:00:00 2001 From: Jianlin Shi Date: Thu, 6 Feb 2025 22:02:47 -0700 Subject: [PATCH] test --- cosmos_openwebui/Dockerfile | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/cosmos_openwebui/Dockerfile b/cosmos_openwebui/Dockerfile index a78ff61..8f7f3e2 100644 --- a/cosmos_openwebui/Dockerfile +++ b/cosmos_openwebui/Dockerfile @@ -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'])"; \