DanTagGen-alpha

项目简介：DanTagGen-alpha

DanTagGen-alpha，即Danbooru标签生成器，是一个灵感来源于p1atdev的dart项目的新项目。然而，它采用了不同的架构、数据集、格式和训练策略。

模型架构

这款DanTagGen版本是在全新的400M参数LLaMA架构（昵称NanoLLaMA）上进行训练的。由于它基于LLaMA架构，理论上可以在任何LLaMA推理接口中使用。

该项目还提供了转换后的FP16 gguf模型，以及量化的8位和6位gguf模型。建议使用llama.cpp或llama-cpp-python运行这个模型，因为它们将非常快速。

格式

DanTagGen采用了一种特定的格式来输入和输出数据。以下是输入格式的示例：

prompt = f"""
rating: {rating or '<|empty|>'}
artist: {artist.strip() or '<|empty|>'}
characters: {characters.strip() or '<|empty|>'}
copyrights: {copyrights.strip() or '<|empty|>'}
aspect ratio: {f"{aspect_ratio:.1f}" or '<|empty|>'}
target: {'<|' + target + '|>' if target else '<|long|>'}
general: {", ".join(special_tags)}, {general.strip().strip(",")}<|input_end|>
"""

例如，输入可能是这样的：

rating: safe
artist: <|empty|>
characters: <|empty|>
copyrights: <|empty|>
aspect ratio: 1.0
target: <|short|>
general: 1girl, solo, dragon girl, dragon horns, dragon tail<|input_end|>

而输出可能会生成以下内容：

rating: safe
artist: <|empty|>
characters: <|empty|>
copyrights: <|empty|>
aspect ratio: 1.0
target: <|short|>
general: 1girl, solo, dragon girl, dragon horns, dragon tail<|input_end|> open mouth, red eyes, long hair, pointy ears, tail, black hair, chinese clothes, simple background, dragon, hair between eyes, horns, china dress, dress, looking at viewer, breasts