rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactor transformers.js usage

Browse Source
This commit is contained in:
Cohee
2023-09-14 23:12:33 +03:00
parent cb8d9ac71b
commit 0cc048cb64
7 changed files with 167 additions and 171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
default/config.conf
View File

@@ -21,6 +21,8 @@ const extras = {
classificationModel: 'Cohee/distilbert-base-uncased-go-emotions-onnx',
// Image captioning model. HuggingFace ID of a model in ONNX format.
captioningModel: 'Xenova/vit-gpt2-image-captioning',
// Feature extraction model. HuggingFace ID of a model in ONNX format.
embeddingModel: 'Xenova/all-mpnet-base-v2,
};
// Request overrides for additional headers