metadata

license: apache-2.0
tags:
  - pytorch
  - diffusers
  - text-to-image

Chinese Latent Diffusion Model

我们开源了一个中文 Lattent Diffusion 模型（美食）

Github: EasyNLP

from diffusers import StableDiffusionPipeline

model_id = "alibaba-pai/pai-diffusion-food-large-zh"
pipe = StableDiffusionPipeline.from_pretrained(model_id)
pipe = pipe.to("cuda")

prompt = "番茄炒蛋"
image = pipe(prompt).images[0]  
image.save("result.png")