Awesome

text2prompt

This is an extension to make prompt from simple text for Stable Diffusion web UI by AUTOMATIC1111.
Currently, only prompts consisting of some danbooru tags can be generated.

Installation

Extensions tab on WebUI

Copy https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git into "Install from URL" tab and "Install".

Install Manually

To install, clone the repository into the extensions directory and restart the web UI.
On the web UI directory, run the following command to install:

git clone https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git extensions/text2prompt

Usage

1. Type some words into "Input Theme"
2. Type some unwanted words into "Input Negative Theme"
3. Push "Generate" button

Tips

• For more creative result
  • increase "k value" or "p value"
  • disable "Use weighted choice"
  • use "Cutoff and Power" and decrease "Power"
  • or use "Softmax" (may generate unwanted tags more often)
• For more strict result
  • decrease "k value" or "p value"
  • use "Cutoff and Power" and increase "Power"
• You can enter very long sentences, but the more specific it is, the fewer results you will get.

How it works

It's doing nothing special;

1. Danbooru tags and it's descriptions are in the data folder
   • embeddigs of descriptions are generated from wiki
   • all-mpnet-base-v2 and all-MiniLM-L6-v2 models are used to make embeddings from the text
2. Tokenize your input text and calculate cosine similarity with all tag descriptions
3. Choose some tags depending on their similarities

Database (Optional)

You can choose the following dataset if needed.
Download the following, unzip and put its contents into text2prompt-root-dir/data/danbooru/.

well filtered: Tags are removed if their description include the title of some work. These tags are heavily related to a specific work, meaning they are not "general" tags.
normal: Tags containing the title of a work, like tag_name(work_name), are removed.
full: Including all tags.

More detailed description

$i \in N = \{1, 2, ..., n\}$ for index number of the tag
$s_i = S_C(d_i, t)$ for cosine similarity between tag description $d_i$ and your text $t$
$P_i$ for probability for the tag to be chosen

"Method to convert similarity into probability"

"Cutoff and Power"

$$p_i = \text{clamp}(s_i, 0, 1)^{\text{Power}} = \text{max}(s_i, 0)^{\text{Power}}$$

"Softmax"

$$p_i = \sigma(\{s_n|n \in N\})i = \dfrac{e^{s_i}}{ \Sigma{j \in N}\ e^{s_j} }$$

"Sampling method"

Yes, it doesn't sample like other "true" language models do, so "Filtering method" might be better.

"NONE"

$$P_i = p_i$$

"Top-k"

$$ P_i = \begin{cases} \dfrac{p_i}{\Sigma p_j \text{ for all top-}k} & \text{if } p_i \text{ is top-}k \text{ largest in } \{p_n | n \in N \} \ 0 & \text{otherwise} \ \end{cases} $$

"Top-p (Nucleus)"

• Find smallest $N_p \subset N$ such that $\Sigma_{i \in N_p}\ p_i\ \geq p$
  • set $N_p=\emptyset$ at first, and add index of $p_{(k)}$ into $N_p$ where $p_{(k)}$ is the $k$-th largest in $\{p_n | n \in N \}$ for $k = 1, 2, ..., n$, until the equation holds.

$$ P_i = \begin{cases} \dfrac{p_i}{\Sigma p_j \text{ for all }j \in N_p} & \text{if } i \in N_p \ 0 & \text{otherwise} \ \end{cases} $$

Finally, the tags will be chosen randomly while the number $\leq$ "Max number of tags".