Stable Diffusion web UI扩展-提示生成器

项目地址：https://github.com/AUTOMATIC1111/stable-diffusion-webui-promptgen

可让您生成提示。

在线获取更多模型

您可以将模特从拥抱脸添加到设置中的模特选择。设置字段 是 ，其默认值为 只：Hugginface model names for promptgen, separated by comma

AUTOMATIC/promptgen-lexart, AUTOMATIC/promptgen-majinai-safe, AUTOMATIC/promptgen-majinai-unsafe

这些是我在各种数据集上所做的 GPT2 微调：

• AUTOMATIC/promptgen-lexart：根据从 lexica.art 抓取的提示微调 distilgpt2 100 个epoch

• AUTOMATIC/promptgen-majinai-safe：在从 majinai.art 抓取的安全提示上微调 distilgpt2 40 个纪元。

• AUTOMATIC/promptgen-majinai-unsafe：根据从 majinai.art 抓取的不安全提示微调 distilgpt2 40 个纪元。

要添加更多模型，请向设置添加更多项目，例如：

AUTOMATIC/promptgen-lexart, AUTOMATIC/promptgen-majinai-safe, AUTOMATIC/promptgen-majinai-unsafe, succinctly/text2image-prompt-generator, microsoft/Promptist

需要重新加载 UI 才能应用此设置。

让更多模型离线

将您的模型放入扩展内的导演中。例如，我的模型可以是 通过从 https://huggingface.co/AUTOMATIC/promptgen-lexart/tree/main 处罚款到完全离线 这些目录：models

 📁 webui root directory
 ┗━━ 📁 extensions
     ┗━━ 📁 stable-diffusion-webui-promptgen
         ┗━━ 📁 models                            
             ┗━━ 📁 promptgen-lexart              <----- any name can be used
                 ┣━━ 📄 config.json               <----- each model has its own set of required files;
                 ┣━━ 📄 merges.txt                       mine requires all those
                 ┣━━ 📄 pytorch_model.bin
                 ┣━━ 📄 tokenizer_config.json
                 ┗━━ 📄 vocab.json

需要重新加载 UI 才能查看新模型。