From 2de55647845dfba64dec78adca3fdb7ff9839b5d Mon Sep 17 00:00:00 2001
From: billwuhao <billwuhao.2018@gmail.com>
Date: Thu, 8 May 2025 13:45:26 +0800
Subject: [PATCH] fix bug

---
 model/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/__init__.py b/model/__init__.py
index 7b09c41..202b11b 100644
--- a/model/__init__.py
+++ b/model/__init__.py
@@ -3,4 +3,4 @@ from model.dit import DiT
 from model.trainer import Trainer
 
 
-__all__ = ["CFM"]
+__all__ = ["CFM", "DiT", "Trainer"]