File size: 150 Bytes
bcc7abd
 
 
 
1
2
3
4
5
from diffusers import DiffusionPipeline
import torch

pipeline = DiffusionPipeline.from_pretrained("segmind/small-sd", torch_dtype=torch.float32)