Update README.md

This commit is contained in:
alpertunga-bile
2024-01-25 21:45:51 +03:00
committed by GitHub
parent a920871a2b
commit 46af8daab0
+12 -7
View File
@@ -24,6 +24,7 @@ Custom AI prompt generator node for [ComfyUI](https://github.com/comfyanonymous/
- [Automatic Installation](#automatic-installation)
- [For Manual Installation of the ComfyUI](#for-manual-installation-of-the-comfyui-2)
- [For Portable Installation of the ComfyUI](#for-portable-installation-of-the-comfyui-1)
- [New Updates On The Node](#new-updates-on-the-node)
- [Contributing](#contributing)
- [Example Outputs](#example-outputs)
@@ -76,10 +77,10 @@ Custom AI prompt generator node for [ComfyUI](https://github.com/comfyanonymous/
## Dataset
- 1.434.667 rows of unique prompts **(Gathering In Process)**
- 1.516.819 rows of unique prompts | **(Gathering In Process)**
- %85 train | %15 test
- Process of data cleaning and gathering can be found [here](https://github.com/alpertunga-bile/prompt-markdown-parser/blob/master/CLI/CLICivitai.py)
- These Huggingface datasets are used:
- These Huggingface datasets are used for enhance the dataset:
- [FredZhang7/anime-prompts-180K](https://huggingface.co/datasets/FredZhang7/anime-prompts-180K)
- [FredZhang7/stable-diffusion-prompts-2.47M](https://huggingface.co/datasets/FredZhang7/stable-diffusion-prompts-2.47M)
- [thefcraft/civitai-stable-diffusion-337k](https://huggingface.co/datasets/thefcraft/civitai-stable-diffusion-337k)
@@ -89,11 +90,11 @@ Custom AI prompt generator node for [ComfyUI](https://github.com/comfyanonymous/
## Models
- female_positive_generator_v2 **(Training In Process)**
- female_positive_generator_v2 | **(Training In Process)**
- using [distilgpt2](https://huggingface.co/distilgpt2) model
- Training Loss ~0.50
- female_positive_generator_v3 **(Training In Process)**
- female_positive_generator_v3 | **(Training In Process)**
- using [bigscience/bloom-560m](https://huggingface.co/bigscience/bloom-560m) model
- Training loss ~0.59
@@ -128,12 +129,12 @@ Custom AI prompt generator node for [ComfyUI](https://github.com/comfyanonymous/
- Look [this link](https://huggingface.co/docs/transformers/v4.31.0/en/generation_strategies#text-generation-strategies) for text generation strategies
## How Recursive Works?
- Let's say we give ```a, ``` as seed and recursive level is 1. I am going to use the same outputs for this example to understand the functionality more accurately.
- Let's say we give ```a, ``` as seed and recursive level is 1. I am going to use the same outputs for this example to explain the functionality more accurately.
- With self recursive, let's say generator's output is ```b```. So next seed is going to be ```b``` and generator's output is ```c```. Final output is ```a, c```. It can be used for generating random outputs.
- Without self recursive, let's say generator's output is ```b```. So next seed is going to be ```a, b``` and generator's output is ```c```. Final output is ```a, b, c```. It can be used for more accurate prompts.
## How Preprocess Mode Works?
- **exact_keyword** => ```(masterpiece), ((masterpiece))``` is not allowed. Checking the pure keyword without parantheses and weights. The algorithm is adding prompts from the beginning of the generated text so add important prompts to seed.
- **exact_keyword** => ```(masterpiece), ((masterpiece))``` is not allowed. Checking the pure keyword without parantheses and weights. The algorithm is adding the prompts from the beginning of the generated text, so add important prompts to seed.
- **exact_prompt** => ```(masterpiece), ((masterpiece))``` is allowed but ```(masterpiece), (masterpiece)``` is not. Checking the exact match of the prompt.
- **none** => Everything is allowed even the repeated prompts.
### Example
@@ -176,13 +177,17 @@ Custom AI prompt generator node for [ComfyUI](https://github.com/comfyanonymous/
- The users have to check that they are starting the ComfyUI in the ```ComfyUI_windows_portable```
- Because the node is checking the ```python_embeded``` folder if it is exists and is using it to install the required packages
## New Updates On The Node
- Sometimes the variables are changed with updates, so it may broke the workflow. But don't worry you have to just delete the node in the workflow and add it again.
# Contributing
- Contributions are welcome. If you have an idea and want to implement it by yourself please follow these steps:
1. Create a fork
2. Create a branch with a name that describes the feature that you are adding
3. Pull request the fork with explaning the new feature
3. Pull request the fork with the comment that explaining the new feature
- If you have an idea but don't know how to implement it please create an issue with ```enhancement``` label.