Gregniuki commited on
Commit
1043b5a
·
verified ·
1 Parent(s): 218ae0c

Upload 4 files

Browse files
model/backbones/dit.py CHANGED
@@ -15,7 +15,7 @@ import torch.nn.functional as F
15
 
16
  from x_transformers.x_transformers import RotaryEmbedding
17
 
18
- from model.modules import (
19
  TimestepEmbedding,
20
  ConvNeXtV2Block,
21
  ConvPositionEmbedding,
 
15
 
16
  from x_transformers.x_transformers import RotaryEmbedding
17
 
18
+ from f5_tts.model.modules import (
19
  TimestepEmbedding,
20
  ConvNeXtV2Block,
21
  ConvPositionEmbedding,
model/backbones/mmdit.py CHANGED
@@ -14,7 +14,7 @@ from torch import nn
14
 
15
  from x_transformers.x_transformers import RotaryEmbedding
16
 
17
- from model.modules import (
18
  TimestepEmbedding,
19
  ConvPositionEmbedding,
20
  MMDiTBlock,
 
14
 
15
  from x_transformers.x_transformers import RotaryEmbedding
16
 
17
+ from f5_tts.model.modules import (
18
  TimestepEmbedding,
19
  ConvPositionEmbedding,
20
  MMDiTBlock,
model/backbones/unett.py CHANGED
@@ -17,7 +17,7 @@ import torch.nn.functional as F
17
  from x_transformers import RMSNorm
18
  from x_transformers.x_transformers import RotaryEmbedding
19
 
20
- from model.modules import (
21
  TimestepEmbedding,
22
  ConvNeXtV2Block,
23
  ConvPositionEmbedding,
 
17
  from x_transformers import RMSNorm
18
  from x_transformers.x_transformers import RotaryEmbedding
19
 
20
+ from f5_tts.model.modules import (
21
  TimestepEmbedding,
22
  ConvNeXtV2Block,
23
  ConvPositionEmbedding,