Superprompt V1

🚀 超級提示器（SuperPrompt）

超級提示器（SuperPrompt）是一個基於T5架構的模型，它在SuperPrompt數據集上進行了微調，能夠將文本提示擴展為更詳細的描述。該模型可作為文本到圖像模型的預生成步驟，為那些受益於更詳細提示的模型提供支持。

🚀 快速開始

安裝依賴

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模型的最大提示長度。

💻 使用示例

示例數據

示例提示取自 DrawBench 數據集。

📄 許可證

本項目採用MIT許可證。