🚀 超级提示器（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许可证。