From e2dd7bfb4b582aeef33a55f7766740d439afd5ab Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Tue, 1 Oct 2024 02:15:27 +0300
Subject: [PATCH] Remove leftovers from prompt expansion

---
 default/config.yaml  | 1 -
 src/transformers.mjs | 6 ------
 2 files changed, 7 deletions(-)

diff --git a/default/config.yaml b/default/config.yaml
index 585988686..a7e7a747e 100644
--- 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 --
diff --git a/src/transformers.mjs b/src/transformers.mjs
index fe947123a..09714e3bb 100644
--- 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,