|
--- |
|
license: apache-2.0 |
|
tags: |
|
- pytorch |
|
- diffusers |
|
- text-to-image |
|
--- |
|
|
|
# LoRA of Chinese Diffusion Model (Poem) |
|
|
|
## 简介 Brief Introduction |
|
|
|
我们开源了适配中文 Diffusion 模型 `alibaba-pai/pai-diffusion-artist-large-zh` 的 LoRA，您可以输入中国古诗词，模型将会根据诗句生成古风水墨配图。 |
|
|
|
We release a LoRA for our Chinese diffusion model `alibaba-pai/pai-diffusion-artist-large-zh`. This model can generate Chinese ink paintings according to the poem you input. |
|
|
|
* Github: [EasyNLP](https://github.com/alibaba/EasyNLP) |
|
|
|
## 使用 Usage |
|
|
|
```python |
|
from diffusers import StableDiffusionPipeline |
|
|
|
model_id = "alibaba-pai/pai-diffusion-artist-large-zh" |
|
pipe = StableDiffusionPipeline.from_pretrained(model_id) |
|
pipe.unet.load_attn_procs("alibaba-pai/pai-diffusion-artist-large-zh-lora-poem") |
|
pipe = pipe.to("cuda") |
|
|
|
prompt = "大漠孤烟直，长河落日圆" |
|
image = pipe(prompt, cross_attention_kwargs={"scale": 0.4}).images[0] |
|
image.save("result.png") |
|
``` |
|
|
|
## 作品展示 Gallery |
|
|
|
| prompt: 大漠孤烟直，长河落日圆 | prompt: 接天莲叶无穷碧，映日荷花别样红 | |
|
| ---------------------------------------- | ---------------------------------- | |
|
| scale: 0.6 | scale: 0.4 | |
|
|  |  | |
|
|
|
## 使用须知 Notice for Use |
|
|
|
使用上述模型需遵守[AIGC模型开源特别条款](https://terms.alicdn.com/legal-agreement/terms/common_platform_service/20230505180457947/20230505180457947.html)。 |
|
|
|
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. |
|
|
|
|
