import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("alibaba-pai/pai-diffusion-general-large-zh", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Chinese Latent Diffusion Model
我们开源了一个中文 Lattent Diffusion 模型（开放领域通用版本）
- Github: EasyNLP
from diffusers import StableDiffusionPipeline
model_id = "alibaba-pai/pai-diffusion-general-large-zh"
pipe = StableDiffusionPipeline.from_pretrained(model_id)
pipe = pipe.to("cuda")
prompt = "雾蒙蒙的日出在湖面上"
image = pipe(prompt).images[0]
image.save("result.png")
- Downloads last month
- 31