https://github.com/p1atdev/danbooru-tags-transformer

https://github.com/p1atdev/danbooru-tags-transformer

Last synced: 6 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/p1atdev/danbooru-tags-transformer
• Owner: p1atdev
• License: apache-2.0
• Created: 2024-02-16T21:18:04.000Z (11 months ago)
• Default Branch: main
• Last Pushed: 2024-10-29T11:54:46.000Z (2 months ago)
• Last Synced: 2024-10-29T13:33:41.736Z (2 months ago)
• Language: Jupyter Notebook
• Size: 593 KB
• Stars: 5
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Danbooru Tags Transformer

Dart (**Da**nboo**r**u **T**ags Transformer) is a large language model that generates prompts for text-to-image models.

SD WebUI extension is [here](https://github.com/p1atdev/sd-danbooru-tags-upsampler). Not this repository.

## v2

|Name|Model architecture|Param size|Model type|Available on|

|:-|-|-|-|-|

|v2-moe-sft|Mixtral|166M|SFT model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v2-moe-sft)|

|v2-moe-base|Mixtral|166M|Pretrained model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v2-moe-base)|

|v2-sft|Mistral|114M|SFT model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v2-sft)|

|v2-base|Mistral|114M|Pretrained model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v2-base)|

|v2-vectors|-|-|Tag Embedding|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v2-vectors)|

## v1

|Model type|Available on|

|:-|-|

|SFT model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v1-sft)|

|Pretrained model|[🤗 HuggingFace](https://huggingface.co/p1atdev/dart-v1-base)|

## Related project(s)

- [DanTagGen](https://huggingface.co/collections/KBlueLeaf/dantaggen-65f82fa9335881a67573556b) by KBlueLeaf