Thuchk's picture
deploy to huggingface spaces
98a48b9
from diffusers import StableDiffusionPipeline
import torch
model_path = "sd-pokemon-model"
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
pipe.to("cuda")
image = pipe(prompt="yoda").images[0]
image.save("yoda-pokemon.png")