Views
No views yet

1from diffusers import DiffusionPipeline, UNet2DConditionModel
2import torch
3
4pipe = DiffusionPipeline.from_pretrained(
5 "stabilityai/stable-diffusion-xl-base-1.0",
6 torch_dtype=torch.float16,
7 unet=UNet2DConditionModel.from_pretrained("pixelparty/pixel-party-xl", torch_dtype=torch.float16),
8 use_safetensors=True,
9 variant="fp16",
10)
11pipe.to("cuda")
12
13torch.manual_seed(11215)
14prompt = "cute dragon. in pixel art style"
15negative_prompt = "mixels. amateur. multiple"
16
17image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=25).images[0]