inference/__init__.py

from .config import (
    ColorCorrectionConfig,
    DecoderConfig,
    HYBRID_B25_ORIGINAL_LAYERS,
    HYBRID_B25_REPLACEMENT_LAYERS,
    InferenceConfig,
    StreamConfig,
)
from .loading import ComponentFactories, ModelBundle, ModelPaths
from .pipeline import FlashVSRInferencePipeline, build_inference_pipeline
from .hardware import HardwareCapabilities, detect_hardware, select_kernels
from .presets import (
    INFERENCE_MODES,
    checkpoint_dir,
    model_paths_for_mode,
    stream_config_for_mode,
)
from .request import InferenceRequest, InferenceResult

__all__ = [
    "ColorCorrectionConfig", "ComponentFactories", "DecoderConfig",
    "FlashVSRInferencePipeline",
    "HardwareCapabilities", "detect_hardware", "select_kernels",
    "HYBRID_B25_ORIGINAL_LAYERS", "HYBRID_B25_REPLACEMENT_LAYERS",
    "INFERENCE_MODES",
    "InferenceConfig", "InferenceRequest", "InferenceResult", "ModelBundle",
    "ModelPaths", "StreamConfig", "build_inference_pipeline", "checkpoint_dir",
    "model_paths_for_mode", "stream_config_for_mode",
]