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, without clothes, blonde hair, open mouth, earrings, beach background, hat, hoop earrings, jewelry, looking at viewer, without shirt, short hair, simple background, solo, upper body, without shirt" with autocast("cuda"): image = pipe(prompt, guidance_scale=6)["sample"][0] image.save("test.png")
#32
by
Eireyn35
- opened
