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

README.md の更新 

wrong terms
main
toshiaki1729 2024-05-24 12:24:00 +09:00 committed by GitHub
parent c4d8cec126
commit bcfec2f988
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

@ -43,8 +43,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
- descriptions are generated from wiki and already tokenized
- [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 tokenize the text
-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