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 2024-05-28 13:10:35 +09:00 committed by GitHub
parent bcfec2f988
commit 752a660d2b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -42,8 +42,8 @@ git clone https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git
It's doing nothing special;
1. Danbooru tags and it's descriptions are in the `data` folder
-embeddigs of descriptions are generated from wiki
1. Danbooru tags and it's descriptions are in the `data` folder
- embeddigs of descriptions are generated from wiki
- [all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) and [all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) models are used to make embeddings from the text
1. Tokenize your input text and calculate cosine similarity with all tag descriptions
1. Choose some tags depending on their similarities