/
githubmirror
/
text-generation-webui
Обзор
Документация
Войти
/
githubmirror
/
text-generation-webui
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
modules/api/cache_embedding_model.py
11 строк
454 B
oobabooga
API: Move OpenAI-compatible API from extensions/openai to modules/api
20 мар 2026, 20:46
20 мар 2026, 20:46
bf6fbc0
Код
Авторство
О чём код?
#!/usr/bin/env python3 # preload the embedding model, useful for Docker images to prevent re-download on config change # Dockerfile: # ENV OPENEDAI_EMBEDDING_MODEL="sentence-transformers/all-mpnet-base-v2" # Optional # RUN python3 cache_embedded_model.py import os import sentence_transformers st_model = os.environ.get("OPENEDAI_EMBEDDING_MODEL", "sentence-transformers/all-mpnet-base-v2") model = sentence_transformers.SentenceTransformer(st_model)