model/__init__.py

# model/__init__.py
from model.config import ModelConfig, SLLM_100M, SLLM_150M
from model.model  import SLLM

__all__ = ["ModelConfig", "SLLM_100M", "SLLM_150M", "SLLM"]