model_lib/__init__.py

# Moebius student
from .nets.unet_lambda_prune_lite import UNet2DLambdaDWConvMixFFNConditionModel_prune_down_mid_up_block_8x8
from .nets.unet_lambda_dwconv_blocks import *  # block factories for student

# PixelHacker teacher (GLA / flash-linear-attention) is training-only; not needed for inference.