SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-01-26 17:20:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

add clarifying progress message

Browse Source
This commit is contained in:
Juha Jeronen
2024-01-24 15:44:33 +02:00
parent 0e506b1ce8
commit 09c007b96c
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
server.py
View File

@@ -601,6 +601,7 @@ def api_embeddings_compute():
sentences: Union[str, List[str]] = data["text"] sentences: Union[str, List[str]] = data["text"]
if not (isinstance(sentences, str) or (isinstance(sentences, list) and all(isinstance(x, str) for x in sentences))): if not (isinstance(sentences, str) or (isinstance(sentences, list) and all(isinstance(x, str) for x in sentences))):
abort(400, '"text" must be string or array of strings') abort(400, '"text" must be string or array of strings')
print("Computing vector embedding")
vectors: Union[np.array, List[np.array]] = sentence_embedder.encode(sentences, vectors: Union[np.array, List[np.array]] = sentence_embedder.encode(sentences,
show_progress_bar=True, # on ST-extras console show_progress_bar=True, # on ST-extras console
convert_to_numpy=True, convert_to_numpy=True,