license: mit

模型卡片：UnfilteredAI/Promt-generator

模型概述

UnfilteredAI/Promt-generator 是一款专为文本到图像模型生成提示而设计的文本生成模型。它采用 PyTorch 和 safetensors 技术，确保在性能和存储方面得到优化，从而能够轻松部署和扩展，适用于提示生成任务。

预期用途

该模型主要适用于以下场景：

• 为文本到图像模型生成提示。
• 在需要生成高质量、多样化图像描述的创意AI应用中发挥关键作用。
• 支持从事生成艺术项目的AI艺术家和开发者。

使用方法

使用该模型生成提示的步骤如下：

1. 在PyTorch环境中加载模型。
2. 输入提示生成任务所需的参数。
3. 模型将根据输入返回文本描述，这些描述可用于文本到图像模型。

示例代码：

from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("UnfilteredAI/Promt-generator")
model = AutoModelForCausalLM.from_pretrained("UnfilteredAI/Promt-generator")

prompt = "a red car"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs)
generated_prompt = tokenizer.decode(outputs[0], skip_special_tokens=True)

print(generated_prompt)