Zonos-main/zonos/backbone/__init__.py

BACKBONES = {}

try:
    from ._mamba_ssm import MambaSSMZonosBackbone

    BACKBONES["mamba_ssm"] = MambaSSMZonosBackbone
except ImportError:
    pass

from ._torch import TorchZonosBackbone

BACKBONES["torch"] = TorchZonosBackbone