1580158
/
stable-diffusion-webui-Prompt_Generator
mirror of https://github.com/imrayya/stable-diffusion-webui-Prompt_Generator.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
generate a prompt from a small base prompt using distilgpt2. Adds a tab with additional control of the model.
25 Commits
1 Branch
0 Tags
75 KiB
Python 88.7%
JavaScript 11.3%
 
 
Go to file
Imrayya 5358234f55 Added Magic Prompt by Gustavosta 2023-01-15 16:25:58 +01:00
scripts Added Magic Prompt by Gustavosta 2023-01-15 16:25:58 +01:00
.gitignore first commit 2022-12-29 01:40:34 +01:00
README.md Added Magic Prompt by Gustavosta 2023-01-15 16:25:58 +01:00
blacklist.txt Added ability to use a blacklist 2023-01-07 19:04:21 +01:00
install.py Doesn't run pip when already installed 2023-01-11 17:55:02 +01:00
licence Added Magic Prompt by Gustavosta 2023-01-15 16:25:58 +01:00

README.md

Prompt Generator

Adds a tab to the webui that allows the user to generate a prompt from a small base prompt. Based on FredZhang7/distilgpt2-stable-diffusion-v2 and Gustavosta/MagicPrompt-Stable-Diffusion. I did nothing apart from porting it to AUTOMATIC1111 WebUI

image

Installation

  1. Install AUTOMATIC1111's Stable Diffusion Webui
  2. Clone this repository into the extensions folder inside the webui

Usage

  1. Write in the prompt in the Start of the prompt text box
  2. Click Generate and wait

Parameters Explanation

  • Start of the prompt: As the name, the start of the prompt that the generator should start with
  • Temperature: A higher temperature will produce more diverse results, but with a higher risk of less coherent text
  • Top K: Strategy is to sample from a shortlist of the top K tokens. This approach allows the other high-scoring tokens a chance of being picked.
  • Max Length: the maximum number of tokens for the output of the model
  • Repetition Penalty: The parameter for repetition penalty. 1.0 means no penalty. See this paper for more details. Default setting is 1.2
  • How Many To Generate: The number of results to generate
  • Use blacklist? Using .\extensions\stable-diffusion-webui-Prompt_Generator\blacklist.txt. It will delete any matches to the generated result (case insensitive). Each item to be filtered out should be on a new line. Be aware that it simply deletes it and doesn't generate more to make up for the lost words

Models

There are two models provided:

FredZhang7

Made by FredZhang7 under creativeml-openrail-m license. Useful to get styles for a prompt. Eg: "A cat sitting" -> "A cat sitting on a chair, digital art. The room is made of clay and metal with the sun shining through in front trending at Artstation 4k uhd..."

MagicPrompt

Made by Gustavosta under the MIT license. Useful to get more natural language prompts. Eg: "A cat sitting" -> "A cat sitting in a chair, wearing pair of sunglasses"

Be aware that sometimes the model fails to produce anything or less than the wanted amount, either try again or use a new prompt in that case

Credits

Credits to both FredZhang7 and Gustavosta