fix bug
This commit is contained in:
+1
-1
@@ -3,4 +3,4 @@ from model.dit import DiT
|
||||
from model.trainer import Trainer
|
||||
|
||||
|
||||
__all__ = ["CFM"]
|
||||
__all__ = ["CFM", "DiT", "Trainer"]
|
||||
|
||||
Reference in New Issue
Block a user