FlexiBrain / flexibrain /models /__init__.py
OneMore1's picture
Sync from GitHub FlexiBrain main
6a51385 verified
Raw
History Blame Contribute Delete
219 Bytes
from flexibrain.models.factory import build_downstream_model, build_pretrain_model
from flexibrain.models.mamba_jepa import VolumeMambaJEPA
__all__ = [build_downstream_model, build_pretrain_model, VolumeMambaJEPA]