File size: 471 Bytes
166ab04 | 1 2 3 4 5 6 7 8 9 10 11 | # 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. The Gradio Space only needs the Moebius student path;
# keep this optional so inference does not require flash-linear-attention.
try:
from .nets.unet_gla import UNet2DGLAConditionModel
except ImportError:
UNet2DGLAConditionModel = None
|