Anime Anything Promptgen V2

🚀 快速動漫提示生成器

快速動漫提示生成器是一個專注於動漫領域的提示生成模型。它基於大量安全的動漫提示數據進行訓練，能夠為用戶生成高質量、富有創意的動漫相關提示，助力動漫創作。

🚀 快速開始

本模型在包含 80,000 條安全動漫提示的數據集上進行了 3 個輪次的訓練。這些提示是從 Safebooru API 端點 獲取的，但僅接受 點贊數 ≥ 8 且不包含任何 黑名單標籤 的唯一提示。 由於 V1 模型經常生成無意義的提示，因此未發佈該版本。在嘗試各種方法糾正這一問題後，最終發現無意義提示的原因並非來自管道參數、模型結構或訓練時長，而是訓練數據中的隨機用戶名。 完整的 提示預處理算法 可參考此處。

✨ 主要特性

• 高質量訓練數據：使用精心篩選的 80,000 條安全動漫提示進行訓練，確保生成提示的質量。
• 有效解決問題：通過深入分析，找出並解決了模型生成無意義提示的問題。

📦 安裝指南

pip install --upgrade transformers

💻 使用示例

基礎用法

import torch
from transformers import GPT2Tokenizer, GPT2LMHeadModel, pipeline
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
model = GPT2LMHeadModel.from_pretrained('FredZhang7/anime-anything-promptgen-v2')

prompt = r'1girl, genshin'

# generate text using fine-tuned model
nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)

# generate 10 samples using contrastive search
outs = nlp(prompt, max_length=76, num_return_sequences=10, do_sample=True, repetition_penalty=1.2, temperature=0.7, top_k=4, early_stopping=True)

print('\nInput:\n' + 100 * '-')
print('\033[96m' + prompt + '\033[0m')
print('\nOutput:\n' + 100 * '-')
for i in range(len(outs)):
    # remove trailing commas and double spaces
    outs[i] = str(outs[i]['generated_text']).replace('  ', '').rstrip(',')
print('\033[92m' + '\n\n'.join(outs) + '\033[0m\n')

高級用法

在上述基礎用法的基礎上，你可以根據自己的需求調整 max_length、num_return_sequences、repetition_penalty、temperature 和 top_k 等參數，以獲得不同風格和數量的生成結果。

📚 詳細文檔

文本到圖像示例

前綴 1girl | 生成的 1girl 提示 | 模型 Anything V4

前綴 1boy  | 生成的 1boy 提示 | 模型 Anything V4

對比搜索

通過對比搜索可以生成更符合需求的文本，具體代碼示例見上方“使用示例”部分。

輸出示例

更多關於管道參數的信息，請參考 快速 GPT 提示生成器。

實用技巧

• 如果你覺得生成的動漫角色缺乏情感，可以在提示中嘗試使用表情符號，如 ;o、:o、;p、:d、:p 和 ;d。也可以使用 happy smirk、happy smile、laughing closed eyes 等描述，讓角色更加生動和富有表現力。
• 在提示中添加 absurdres 而不是 highres 和 masterpiece，可以顯著提高生成圖像的清晰度和分辨率。

Danbooru 版本

Danbooru 版本鏈接

📄 許可證

本項目採用 CreativeML OpenRAIL-M 許可證。

📋 信息表格