---

language: en

tags:

- tagforge

- danbooru

- image-tagging

- comfyui

license: gpl-3.0

---

TagForge-0.4B

Version 0.7

Decoder-only transformer that generates structured Danbooru tag sets for

single-character images. It adheres mostly to Danbooru styling tags, so it generates

outfits based on actual, working outfit mixes — describe a style (if known) or just part

of an outfit, and it completes a coherent character/clothing set across every body zone. Supports NSFW tags fully, but only designed to handle a solo character - interaction tags will be handled by an intelligent module in the node pack.

Part of the TagForgeSuite ComfyUI node pack, but usable in any LLM inferencing tool.

Architecture

| Parameter | Value |

|--------------------|------------|

| Vocab size | 153,728 |

| Embedding dim | 1024 |

| Transformer layers | 16 |

| Attention heads | 16 |

| FFN dim | 4,096 |

| Parameters | ~361M |

| Dtype | bfloat16 |

The embedding and output head are tied (one token per tag).

Usage

Load via the TagForge Loader node in TagForgeSuite. The node uses the original-format

tagforge_tokenizer.json (included). For raw transformers use, tokenizer.json is the

HuggingFace WordLevel tokenizer.

Commands: GEN, FILL, EDIT to generate; CLASSIFY to sort tags into their body/scene zones.

Example — [GEN] 1girl, tomboy, petite, overalls, small_breasts, red_hair, pixie_cut, blue_eyes, t-shirt

might output:

RATING : s | SUBJECT : 1girl , petite , tomboy | BODY : breasts , small_breasts | META : simple_background , white_background | HAIR : pixie_cut , red_hair | FACE : blue_eyes | UPPER_BODY : ATTIRE : overalls , shirt , t-shirt | LOWER_BODY : ATTIRE : shorts | EXPRESSION : blush , looking_at_viewer , smile | ACTION : STATE : standing

Add a RATING in your input (g, s, q, e — as on Danbooru) - like this: "..., RATING : q, ..." and the model adheres to it without dropping the

tags you gave it. g = general/SFW, s = sensitive, q = questionable, e = explicit (very).

License

GPL-3.0