1580158
/
stable-diffusion-webui-text2prompt
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update README.md

main
toshiaki1729 2022-12-25 00:42:25 +09:00 committed by GitHub
parent b3e2d931f6
commit 05051d8064
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -28,7 +28,7 @@ git clone https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git
1. Danbooru tags and it's descriptions are in the `data` folder
- descriptions are generated from wiki and already tokenised
- [all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) model is used to tokenize the text
- for now, some tags (<1k tagged, containing title of the work) are deleted to prevent from "noisy" result
- for now, some tags (such as <1k tagged or containing title of the work) are deleted to prevent from "noisy" result
1. Tokenize your input text and calculate cosine similarity with all tag descriptions
1. Choose some tags depending on their similarities