| import torch |
| from diffusers import ( |
| StableDiffusionXLPipeline, |
| EulerAncestralDiscreteScheduler, |
| AutoencoderKL |
| ) |
|
|
| |
| vae = AutoencoderKL.from_pretrained( |
| "madebyollin/sdxl-vae-fp16-fix", |
| torch_dtype=torch.float16 |
| ) |
|
|
| |
| pipe = StableDiffusionXLPipeline.from_pretrained( |
| "cagliostrolab/animagine-xl-3.0", |
| vae=vae, |
| torch_dtype=torch.float16, |
| use_safetensors=True, |
| ) |
| pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config) |
| pipe.to('cuda') |
|
|
| |
| prompt = "1girl, arima kana, oshi no ko, solo, upper body, v, smile, looking at viewer, outdoors, night" |
| negative_prompt = "nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name" |
|
|
| image = pipe( |
| prompt, |
| negative_prompt=negative_prompt, |
| width=832, |
| height=1216, |
| guidance_scale=7, |
| num_inference_steps=28 |
| ).images[0] |
|
|
