protify/FastPLMs/boltz_fastplms/__init__.py

from boltz_fastplms.modeling_boltz2 import (
    Boltz2Config,
    Boltz2Model,
    Boltz2StructureOutput,
)

__all__ = [
    "Boltz2Config",
    "Boltz2Model",
    "Boltz2StructureOutput",
]