diff --git a/src/boltz/model/models/__init__.py b/src/boltz/model/models/__init__.py
index e69de29bb..8d1493b56 100644
--- a/src/boltz/model/models/__init__.py
+++ b/src/boltz/model/models/__init__.py
@@ -0,0 +1,7 @@
+from .boltz2 import Boltz2
+# (optional) import other models here if needed
+# from .boltz1 import Boltz1
+# from .unet import UNet
+
+__all__ = ["Boltz2"]
+