Superprompt V1

🚀 スーパープロンプト

T5モデルをSuperPromptデータセットでファインチューニングしたもので、テキストプロンプトをより詳細な説明に拡張します。より詳細なプロンプトが有益なテキストから画像への生成モデルの事前生成ステップとして使用することを想定しています。

🚀 クイックスタート

このモデルを使用するには、まず必要なライブラリをインストールする必要があります。以下のコマンドを使用して、transformers ライブラリをインストールします。

pip install transformers

💻 使用例

基本的な使用法

from transformers import T5Tokenizer, T5ForConditionalGeneration

tokenizer = T5Tokenizer.from_pretrained("roborovski/superprompt-v1")
model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1", device_map="auto")

input_text = "Expand the following prompt to add more detail: A storefront with 'Text to Image' written on it."
input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")

outputs = model.generate(input_ids, max_new_tokens=77)
print(tokenizer.decode(outputs[0]))

# The neon sign above the storefront reads "NeurIPS" in bold, white letters. The storefront is surrounded by a bustling cityscape, with skyscrapers and neon signs lining the walls. The sign is surrounded by a variety of colorful goods, including a variety of fruits, vegetables, and fruits, all arranged in a neat and organized manner. The storefront is surrounded by a bustling crowd of people, all chatting and laughing as they go about their daily routines.

重要な注意事項

正確なタスクプレフィックス Expand the following prompt to add more detail: を使用し、最大トークン数を77に制限することが重要です。これはStable Diffusionモデルの最大プロンプト長です。

⚠️ 重要提示

正確なタスクプレフィックス Expand the following prompt to add more detail: を使用し、最大トークン数を77に制限することが重要です。これはStable Diffusionモデルの最大プロンプト長です。

📚 ドキュメント

サンプル

DrawBench データセットから取ったサンプルプロンプトです。

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。