Bavarder/src/providers/stablediffusion.py

6 lines
265 B
Python

from .basehfimage import BaseHFImageProvider
class StableDiffusionProvider(BaseHFImageProvider):
name = "Stable Diffusion"
provider = "stabilityai/stable-diffusion-2-1"
description = "Stable Diffusion is a model that can generate images from a prompt."