Spaces:
Running on Zero
Running on Zero
File size: 1,541 Bytes
b701455 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | """Core module for LightDiffusion-Next.
This module provides the foundational abstractions for the modular pipeline:
- Context: Simplified state container for all pipeline state
- Interfaces: Protocol definitions (ModelProtocol, ProcessorProtocol, etc.)
- Pipeline: Main orchestrator class
- AbstractModel: Base class for model implementations
- Models: Concrete implementations (SD15Model, SDXLModel)
"""
from src.Core.Context import Context, SamplingConfig, GenerationConfig, FeatureFlags
from src.Core.Interfaces import (
ModelCapabilities,
ModelProtocol,
ProcessorProtocol,
BaseProcessor,
SamplerProtocol,
CFGSchedulerProtocol,
ConstantCFGScheduler,
CFGFreeScheduler,
LinearDecayCFGScheduler,
PipelineProtocol,
)
from src.Core.AbstractModel import AbstractModel
from src.Core.Pipeline import Pipeline, PipelineResult, get_default_pipeline
from src.Core.Models import SD15Model, SDXLModel, create_model, detect_model_type
__all__ = [
# Context API
"Context",
"SamplingConfig",
"GenerationConfig",
"FeatureFlags",
# Pipeline
"Pipeline",
"PipelineResult",
"get_default_pipeline",
# Protocols
"ModelCapabilities",
"ModelProtocol",
"ProcessorProtocol",
"BaseProcessor",
"SamplerProtocol",
"CFGSchedulerProtocol",
"ConstantCFGScheduler",
"CFGFreeScheduler",
"LinearDecayCFGScheduler",
"PipelineProtocol",
# Models
"AbstractModel",
"SD15Model",
"SDXLModel",
"create_model",
"detect_model_type",
]
|