Remove leftovers from prompt expansion

e2dd7bfb4b582aeef33a55f7766740d439afd5ab

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: a63f99b8ce3c3f20370f265ad4359088c5876df0
2 files changed, +0 -7Showing whitespace changes
default/config.yaml+0 -1
@@ -118,7 +118,6 @@ extras:
118 classificationModel: Cohee/distilbert-base-uncased-go-emotions-onnx118 classificationModel: Cohee/distilbert-base-uncased-go-emotions-onnx
119 captioningModel: Xenova/vit-gpt2-image-captioning119 captioningModel: Xenova/vit-gpt2-image-captioning
120 embeddingModel: Cohee/jina-embeddings-v2-base-en120 embeddingModel: Cohee/jina-embeddings-v2-base-en
121 promptExpansionModel: Cohee/fooocus_expansion-onnx
122 speechToTextModel: Xenova/whisper-small121 speechToTextModel: Xenova/whisper-small
123 textToSpeechModel: Xenova/speecht5_tts122 textToSpeechModel: Xenova/speecht5_tts
124# -- OPENAI CONFIGURATION --123# -- OPENAI CONFIGURATION --
src/transformers.mjs+0 -6
@@ -31,12 +31,6 @@ const tasks = {
31 configField: 'extras.embeddingModel',31 configField: 'extras.embeddingModel',
32 quantized: true,32 quantized: true,
33 },33 },
34 'text-generation': {
35 defaultModel: 'Cohee/fooocus_expansion-onnx',
36 pipeline: null,
37 configField: 'extras.promptExpansionModel',
38 quantized: false,
39 },
40 'automatic-speech-recognition': {34 'automatic-speech-recognition': {
41 defaultModel: 'Xenova/whisper-small',35 defaultModel: 'Xenova/whisper-small',
42 pipeline: null,36 pipeline: null,