src/designgenie/models/__init__.py

from .diffusion import (
    StableDiffusionControlNet,
    StableDiffusionControlNetInpaint,
    create_diffusion_model,
)
from .segmentation import MaskFormer, Mask2Former, create_segmentation_model