Commit
•
93b99de
1
Parent(s):
736bdcb
add model
Browse files
.ipynb_checkpoints/README-checkpoint.md
DELETED
|
@@ -1,65 +0,0 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
tags:
|
| 4 |
-
- pytorch
|
| 5 |
-
- transformers
|
| 6 |
-
- text-generation
|
| 7 |
-
---
|
| 8 |
-
|
| 9 |
-
# BeautifulPrompt
|
| 10 |
-
|
| 11 |
-
## 简介 Brief Introduction
|
| 12 |
-
|
| 13 |
-
我们开源了一个自动Prompt生成模型，您可以直接输入一个极其简单的Prompt，就可以得到经过语言模型优化过的Prompt，帮助您更简单地生成高颜值图像。
|
| 14 |
-
|
| 15 |
-
We release an automatic Prompt generation model, you can directly enter an extremely simple Prompt and get a Prompt optimized by the language model to help you generate high value images more simply.
|
| 16 |
-
|
| 17 |
-
* Github: [EasyNLP](https://github.com/alibaba/EasyNLP)
|
| 18 |
-
|
| 19 |
-
## 使用 Usage
|
| 20 |
-
|
| 21 |
-
```python
|
| 22 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 23 |
-
|
| 24 |
-
tokenizer = AutoTokenizer.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd')
|
| 25 |
-
model = AutoModelForCausalLM.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd').eval().cuda()
|
| 26 |
-
|
| 27 |
-
raw_prompt = '1 girl'
|
| 28 |
-
input = f'Instruction: Give a simple description of the image to generate a drawing prompt.\nInput: {raw_prompt}\nOutput:'
|
| 29 |
-
input_ids = tokenizer.encode(input, return_tensors='pt').cuda()
|
| 30 |
-
|
| 31 |
-
outputs = model.generate(
|
| 32 |
-
input_ids,
|
| 33 |
-
max_length=384,
|
| 34 |
-
do_sample=True,
|
| 35 |
-
temperature=1.0,
|
| 36 |
-
top_k=50,
|
| 37 |
-
top_p=0.95,
|
| 38 |
-
repetition_penalty=1.2,
|
| 39 |
-
num_return_sequences=5)
|
| 40 |
-
|
| 41 |
-
prompts = tokenizer.batch_decode(outputs[:, input_ids.size(1):], skip_special_tokens=True)
|
| 42 |
-
prompts = [p.strip() for p in prompts]
|
| 43 |
-
print(prompts)
|
| 44 |
-
```
|
| 45 |
-
|
| 46 |
-
## 作品展示 Gallery
|
| 47 |
-
|
| 48 |
-
| Original | BeautifulPrompt |
|
| 49 |
-
| ---------------------------------------- | ---------------------------------- |
|
| 50 |
-
| prompt: taylor swift, country, golden, fearless,wavehair | prompt: portrait of taylor swift as a beautiful woman, long hair, country, golden ratio, intricate, symmetrical, cinematic lighting, highly detailed, digital painting, artstation, concept art, smooth, sharp focus, illustration |
|
| 51 |
-
|  |  |
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
| Original | BeautifulPrompt |
|
| 55 |
-
| ---------------------------------------- | ---------------------------------- |
|
| 56 |
-
| prompt: A majestic sailing ship | prompt: a massive sailing ship, epic, cinematic, artstation, greg rutkowski, james gurney, sparth |
|
| 57 |
-
|  |  |
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
## 使用须知 Notice for Use
|
| 62 |
-
|
| 63 |
-
使用上述模型需遵守[AIGC模型开源特别条款](https://terms.alicdn.com/legal-agreement/terms/common_platform_service/20230505180457947/20230505180457947.html)。
|
| 64 |
-
|
| 65 |
-
If you want to use this model, please read this [document](https://terms.alicdn.com/legal-agreement/terms/common_platform_service/20230505180457947/20230505180457947.html) carefully and abide by the terms.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|