Remove leftovers from prompt expansion

2024-10-01 02:15:27 +03:00 · 2024-10-01 02:15:27 +03:00 · e2dd7bfb4b
parent a63f99b8ce
commit e2dd7bfb4b
2 changed files with 0 additions and 7 deletions
--- a/default/config.yaml
+++ b/default/config.yaml
@ -118,7 +118,6 @@ extras:
  classificationModel: Cohee/distilbert-base-uncased-go-emotions-onnx
  captioningModel: Xenova/vit-gpt2-image-captioning
  embeddingModel: Cohee/jina-embeddings-v2-base-en
-  promptExpansionModel: Cohee/fooocus_expansion-onnx
  speechToTextModel: Xenova/whisper-small
  textToSpeechModel: Xenova/speecht5_tts
 # -- OPENAI CONFIGURATION --
--- a/src/transformers.mjs
+++ b/src/transformers.mjs
@ -31,12 +31,6 @@ const tasks = {
        configField: 'extras.embeddingModel',
        quantized: true,
    },
-    'text-generation': {
-        defaultModel: 'Cohee/fooocus_expansion-onnx',
-        pipeline: null,
-        configField: 'extras.promptExpansionModel',
-        quantized: false,
-    },
    'automatic-speech-recognition': {
        defaultModel: 'Xenova/whisper-small',
        pipeline: null,