src/Core/Models/__init__.py

"""Model adapters for LightDiffusion-Next.

This module provides concrete model implementations that inherit from
AbstractModel and wrap the existing model infrastructure.

These adapters provide a clean interface while reusing the existing
heavily-tested model loading and inference code.
"""

from src.Core.Models.SD15Model import SD15Model
from src.Core.Models.SDXLModel import SDXLModel
from src.Core.Models.Flux2KleinModel import Flux2KleinModel
from src.Core.Models.ModelFactory import (
    create_model,
    detect_model_type,
    detect_model_type_from_state_dict,
    register_model_type,
    list_model_types,
    list_available_models,
)

__all__ = [
    "SD15Model",
    "SDXLModel",
    "Flux2KleinModel",
    "create_model",
    "detect_model_type",
    "detect_model_type_from_state_dict",
    "register_model_type",
    "list_model_types",
    "list_available_models",
]