awesome-tiny-sd / imgen.py
AstraBert
build commit
bcc7abd
from diffusers import DiffusionPipeline
import torch
pipeline = DiffusionPipeline.from_pretrained("segmind/small-sd", torch_dtype=torch.float32)