| import torch | |
| from torch import autocast | |
| from diffusers import StableDiffusionPipeline | |
| pipe = StableDiffusionPipeline.from_pretrained( | |
| 'hakurei/waifu-diffusion', | |
| torch_dtype=torch.float32 | |
| ).to('cuda') | |
| prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt" | |
| with autocast("cuda"): | |
| image = pipe(prompt, guidance_scale=6)["sample"][0] | |
| image.save("test.png") | |