Spaces:
Runtime error
Runtime error
File size: 263 Bytes
98a48b9 |
1 2 3 4 5 6 7 8 9 |
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") |