Views
No views yet
1import os
2from diffusers import StableDiffusionPipeline
3import torch
4import random
5
6prompt="可爱的,黄色的老鼠,小精灵,卡通"
7seed=random.randint(1,2147483647)
8print(seed)
9pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Anime-Chinese-v0.1",
10 torch_dtype=torch.float16,)
11model_path = "miluELK/Taiyi-sd-pokemon-LoRA-zh-512-v2"
12
13pipe.unet.load_attn_procs(model_path)
14pipe.to("cuda")
15generator = torch.Generator("cuda").manual_seed(seed)
16pipe.safety_checker = lambda images, clip_input: (images, False)
17image = pipe(prompt, generator=generator, num_inference_steps=50, guidance_scale=7.5).images[0]
18image.save("./" + prompt + ".png")
19image


