| import torch | |
| from torch import autocast | |
| from diffusers import StableDiffusionPipeline | |
| model_id = "hakurei/waifu-diffusion" | |
| device = "cuda" | |
| pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision='fp16') | |
| pipe = pipe.to(device) | |
| prompt = "touhou hakurei_reimu 1girl solo portrait" | |
| with autocast("cuda"): | |
| image = pipe(prompt, guidance_scale=7.5)["sample"][0] | |
| image.save("reimu_hakurei.png") |