This commit is contained in:
billwuhao
2025-05-08 13:45:26 +08:00
parent 09e43b6e76
commit 2de5564784
+1 -1
View File
@@ -3,4 +3,4 @@ from model.dit import DiT
from model.trainer import Trainer
__all__ = ["CFM"]
__all__ = ["CFM", "DiT", "Trainer"]