text-generation-webui/extensions/openai/cache_embedding_model.py

#!/usr/bin/env python3
# preload the embedding model, useful for Docker images to prevent re-download on config change
# Dockerfile:
# ENV OPENEDAI_EMBEDDING_MODEL=all-mpnet-base-v2 # Optional
# RUN python3 cache_embedded_model.py
import os

import sentence_transformers

st_model = os.environ.get("OPENEDAI_EMBEDDING_MODEL", "all-mpnet-base-v2")
model = sentence_transformers.SentenceTransformer(st_model)
[extension/openai] add edits & image endpoints & fix prompt return in non --chat modes (#1935) 2023-05-11 10:06:39 -04:00			`#!/usr/bin/env python3`
			`# preload the embedding model, useful for Docker images to prevent re-download on config change`
			`# Dockerfile:`
			`# ENV OPENEDAI_EMBEDDING_MODEL=all-mpnet-base-v2 # Optional`
			`# RUN python3 cache_embedded_model.py`
Lint the openai extension 2023-09-15 20:11:16 -07:00			`import os`

			`import sentence_transformers`

extensions/openai: Fix error when preparing cache for embedding models (#3995) 2023-09-24 11:58:28 +08:00			`st_model = os.environ.get("OPENEDAI_EMBEDDING_MODEL", "all-mpnet-base-v2")`
[extension/openai] add edits & image endpoints & fix prompt return in non --chat modes (#1935) 2023-05-11 10:06:39 -04:00			`model = sentence_transformers.SentenceTransformer(st_model)`