Imrayya/stable-diffusion-webui-Prompt_Generator
1
0
Fork 0
mirror of https://github.com/imrayya/stable-diffusion-webui-Prompt_Generator.git synced 2024-01-11 09:00:44 +01:00
Code Issues Releases Wiki Activity

Allow for more arbitrary models

Browse Source
This commit is contained in:
Kareem
2023-02-11 12:06:18 +01:00
parent b4833e90ce
commit 952f3fee41
3 changed files with 70 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+11 -3
View File
@@ -13,7 +13,8 @@ Adds a tab to the webui that allows the user to generate a prompt from a small b
## Usage
1. Write in the prompt in the *Start of the prompt* text box
2. Click Generate and wait
2. Select which model you want to use
3. Click Generate and wait
The initial use of the model may take longer as it needs to be downloaded to your machine for offline use. The model will be used on your device and will be stored in the default location of `*username*/.cache/huggingface/hub/models`. The entire process of generating results will be done on your local machine and not require internet access.
@@ -25,11 +26,12 @@ The initial use of the model may take longer as it needs to be downloaded to you
- **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](https://arxiv.org/pdf/1909.05858.pdf) 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*
- **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*
- **Use puncation**: Allows the use commas in the output
## Models
There are two models provided:
There are two 'default' models provided:
### FredZhang7
@@ -45,6 +47,12 @@ Useful to get more natural language prompts. Eg: "A cat sitting" -> "A cat sitti
*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*
## Install more models
To install more model to use, ensure that the models are hosted on [huggingface.co](https://huggingface.co) and edit the json file at `.\extensions\stable-diffusion-webui-Prompt_Generator\models.json` with the relevant information. Use the models in the file as an basis
You might need to restart the extension/reload the UI if new items are added onto the list
## Credits
Credits to both [FredZhang7](https://huggingface.co/FredZhang7) and [Gustavosta](https://huggingface.co/Gustavosta)