Views
No views yet

1import torch
2from torch import autocast
3from diffusers import StableDiffusionPipeline
4
5pipe = StableDiffusionPipeline.from_pretrained(
6 'AlexWortega/AnimeDiffuion2',
7 torch_dtype=torch.float32
8).to('cuda')
9
10negative_prompt = """low-res, duplicate, poorly drawn face, ugly, undetailed face"""
11d = 'white girl'
12prompt = f"pink hair guy in glasses, photograph, sporty body, cinematic lighting, clear eyes, perfect face, blush, beautiful nose, beautiful eyes, detailed eyes"
13num_samples = 1
14
15with torch.inference_mode():
16 images = pipe([prompt] * num_samples,
17 negative_prompt = [negative_prompt]*num_samples,
18 height=512, width=512,
19 num_inference_steps=50,
20 guidance_scale=8,
21 ).images
22
23
24
25images[0].save("test.png")