Views
No views yet
Girl with a pearl earring, Cute Obama creature, Donald Trump, Boris Johnson, Totoro, Hello Kitty
1!pip install diffusers==0.3.0
2!pip install transformers scipy ftfy1import torch
2from diffusers import StableDiffusionPipeline
3from torch import autocast
4
5pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/sd-pokemon-diffusers", torch_dtype=torch.float16)
6pipe = pipe.to("cuda")
7
8prompt = "Yoda"
9scale = 10
10n_samples = 4
11
12# Sometimes the nsfw checker is confused by the Pokémon images, you can disable
13# it at your own risk here
14disable_safety = False
15
16if disable_safety:
17 def null_safety(images, **kwargs):
18 return images, False
19 pipe.safety_checker = null_safety
20
21with autocast("cuda"):
22 images = pipe(n_samples*[prompt], guidance_scale=scale).images
23
24for idx, im in enumerate(images):
25 im.save(f"{idx:06}.png")