Flexpert-Design/src/models/__init__.py

# Copyright (c) CAIRI AI Lab. All rights reserved

# from .alphadesign_model import AlphaDesign_Model
# from .esmif_model import GVPTransformerModel as ESMIF_Model
# from .gca_model import GCA_Model
# from .graphtrans_model import GraphTrans_Model
# from .gvp_model import GVP_Model
# from .pifold_model import PiFold_Model
from .proteinmpnn_model import ProteinMPNN_Model
# from .structgnn_model import StructGNN_Model
# from .kwdesign_model import KWDesign_model

__all__ = [
    'AlphaDesign_Model', 'ESMIF_Model', 'GCA_Model', 'GraphTrans_Model', 'GVP_Model',
    'PiFold_Model', 'ProteinMPNN_Model', 'StructGNN_Model', 'KWDesign_model'
]