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

Text classify truncation

Browse Source
This commit is contained in:
SillyLossy
2023-04-14 13:50:58 +03:00
parent 28351a4ca8
commit 154a0facb4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server.py
View File

@@ -206,7 +206,7 @@ def require_module(name):
# AI stuff
def classify_text(text: str) -> list:
output = classification_pipe(text)[0]
output = classification_pipe(text, truncation=True, max_length=classification_pipe.model.config.max_position_embeddings)[0]
return sorted(output, key=lambda x: x['score'], reverse=True)