From 31567385f092f8e7fa3dd2d8d2f1a0999651ef53 Mon Sep 17 00:00:00 2001
From: arunodhayan <34075721+arunodhayan@users.noreply.github.com>
Date: Tue, 21 Oct 2025 07:35:23 +0200
Subject: [PATCH] Update __init__.py

---
 src/boltz/model/models/__init__.py | 7 +++++++
 1 file changed, 7 insertions(+)

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"]
+