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

noise2023-08-022024-11-09

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

可让您生成提示。

在线获取更多模型

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

1	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 个纪元。

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

1	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 才能查看新模型。