--- 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 | | ![](example1.png) | ![](example2.png) | ## 使用须知 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.