File size: 442 Bytes
60ea83f b0171e9 60ea83f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# from f5_tts.model.cfm import CFM
#
# from f5_tts.model.backbones.unett import UNetT
from f5_tts.model.backbones.dit import DiT
# from f5_tts.model.backbones.dit import DiTNoCond
# from f5_tts.model.backbones.dit import DiTNoCondNoT
# from f5_tts.model.backbones.mmdit import MMDiT
# from f5_tts.model.trainer import Trainer
# __all__ = ["CFM", "UNetT", "DiT", "MMDiT", "Trainer"]
# __all__ = ["CFM", "UNetT", "DiTNoCond","DiT", "MMDiT"]
|