mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

extensions/openai: Fix error when preparing cache for embedding models (#3995)

Browse source
This commit is contained in:
Chenxiao Wang 2023-09-24 11:58:28 +08:00 • committed by GitHub
parent 7a3ca2c68f
commit 9de2dfa887
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
extensions/openai/cache_embedding_model.py
View file

@ -6,7 +6,6 @@
import os
import sentence_transformers
from extensions.openai.script import params
st_model = os.environ.get("OPENEDAI_EMBEDDING_MODEL", params.get('embedding_model', 'all-mpnet-base-v2'))
st_model = os.environ.get("OPENEDAI_EMBEDDING_MODEL", "all-mpnet-base-v2")
model = sentence_transformers.SentenceTransformer(st_model)