license: apache-2.0 widget:

• text: "指令：对图像进行简单描述以生成绘图提示词\n输入：1个女孩\n输出：" tags:
• pytorch
• transformers
• 文本生成

美颜提示词生成器

简介

我们开源了自动提示词优化模型，只需输入极简描述，即可获得经语言模型优化的专业级提示词，助您轻松生成高质感图像。

• GitHub项目：EasyNLP

使用方法

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd')
model = AutoModelForCausalLM.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd').eval().cuda()

raw_prompt = '1个女孩'
input = f'指令：对图像进行简单描述以生成绘图提示词\n输入：{raw_prompt}\n输出：'
input_ids = tokenizer.encode(input, return_tensors='pt').cuda()

outputs = model.generate(
    input_ids,
    max_length=384,
    do_sample=True,
    temperature=1.0,
    top_k=50,
    top_p=0.95,
    repetition_penalty=1.2,
    num_return_sequences=5)

prompts = tokenizer.batch_decode(outputs[:, input_ids.size(1):], skip_special_tokens=True)
prompts = [p.strip() for p in prompts]
print(prompts)

效果展示

原始提示词	优化后提示词
提示词：泰勒·斯威夫特，乡村风，金色，无畏专辑，波浪发	提示词：泰勒·斯威夫特肖像，绝美女性造型，长发飘逸，乡村风情，黄金比例构图，精致细节，对称美学，电影级光影，超高精度数字绘画，ArtStation艺术平台标准，概念艺术质感，平滑笔触，锐利焦点插画风格

原始提示词	优化后提示词
提示词：雄伟帆船	提示词：巨型帆船航行，史诗级场景，电影镜头感，ArtStation艺术标准，Greg Rutkowski与James Gurney联合风格，Sparth式构图

使用须知

使用本模型需遵守AIGC模型开源特别条款。

文献引用

若使用本模型，请引用以下论文：

@inproceedings{emnlp2023a,
  author    = {曹庭锋 and
            王成宇 and
            刘冰彦 and
            吴子恒 and
            朱金辉 and
            黄俊},
  title     = {美颜提示词：面向文生图合成的自动提示词工程技术},
  booktitle = {2023年自然语言处理实证方法会议：工业应用分册},
  pages     = {1--11},
  year      = {2023}
}