Add SD prompt expansion

2023-10-20 15:03:26 +03:00
parent 5c6343e85e
commit c4e6b565a5
5 changed files with 129 additions and 1 deletions
@@ -26,6 +26,8 @@ const extras = {
    captioningModel: 'Xenova/vit-gpt2-image-captioning',
    // Feature extraction model. HuggingFace ID of a model in ONNX format.
    embeddingModel: 'Xenova/all-mpnet-base-v2',
+    // GPT-2 text generation model. HuggingFace ID of a model in ONNX format.
+    promptExpansionModel: 'Cohee/fooocus_expansion-onnx',
 };

 // Request overrides for additional headers