Gregniuki commited on
Commit
8f8c6f1
1 Parent(s): 8ac0592

Update model/__init__.py

Browse files
Files changed (1) hide show
  1. model/__init__.py +5 -5
model/__init__.py CHANGED
@@ -1,10 +1,10 @@
1
- from f5_tts.model.cfm import CFM
2
 
3
- from f5_tts.model.backbones.unett import UNetT
4
- from f5_tts.model.backbones.dit import DiT
5
- from f5_tts.model.backbones.mmdit import MMDiT
6
 
7
- from f5_tts.model.trainer import Trainer
8
 
9
 
10
  __all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]
 
1
+ from model.cfm import CFM
2
 
3
+ from model.backbones.unett import UNetT
4
+ from model.backbones.dit import DiT
5
+ from model.backbones.mmdit import MMDiT
6
 
7
+ from model.trainer import Trainer
8
 
9
 
10
  __all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]