Model Type | |
Use Cases |
Areas: | Research, Tag generation, Danbooru tag-specific applications |
|
Applications: | Tag generation for images |
|
Primary Use Cases: | Generating Danbooru tags for specified inputs. |
|
Limitations: | Cannot accommodate flexible specifications. |
|
Considerations: | Due to training with a specialized prompt format, natural language is not supported. |
|
|
Supported Languages | Danbooru tags (Primary supported input format) |
|
Training Details |
Data Sources: | isek-ai/danbooru-tags-2023 |
|
Data Volume: | 6M size of danbooru tags dataset |
|
Methodology: | Trained using ๐ค transformers' trainer. |
|
Hardware Used: | |
Model Architecture: | OPT (Open Pretrained Transformer), but the position embeddings were not trained. |
|
|
Input Output |
Input Format: | <|bos|> [RATING_PARENT], [RATING_CHILD] [COPYRIGHT, ...] [CHARACTER, ...] [LENGTH_TOKEN][GENERAL, ...]<|input_end|> [COMPLETION] <|eos|> |
|
Accepted Modalities: | |
Output Format: | Model completes tags in alphabetical order after [COMPLETION]. |
|
|