src/sllm/__init__.py

from sllm.config import DataMixConfig, ModelConfig, SFTConfig, SourceConfig, TrainConfig
from sllm.model import SLLMForCausalLM

__all__ = [
    "DataMixConfig",
    "ModelConfig",
    "SFTConfig",
    "SourceConfig",
    "TrainConfig",
    "SLLMForCausalLM",
]