from . import ben2, da3, moge2, upscale


def geometry_adapter(model_key):
    return moge2 if model_key.startswith("MOGE2_") else da3


__all__ = (
    "ben2",
    "da3",
    "moge2",
    "upscale",
    "geometry_adapter",
)