diff --git a/GPT_SoVITS/f5_tts/model/__init__.py b/GPT_SoVITS/f5_tts/model/__init__.py index 50cff994..bf02a676 100644 --- a/GPT_SoVITS/f5_tts/model/__init__.py +++ b/GPT_SoVITS/f5_tts/model/__init__.py @@ -1,7 +1,7 @@ # from f5_tts.model.cfm import CFM # # from f5_tts.model.backbones.unett import UNetT -from GPT_SoVITS.f5_tts.model.backbones.dit import DiT +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 diff --git a/GPT_SoVITS/f5_tts/model/backbones/dit.py b/GPT_SoVITS/f5_tts/model/backbones/dit.py index 4aa3b9ac..d8073c31 100644 --- a/GPT_SoVITS/f5_tts/model/backbones/dit.py +++ b/GPT_SoVITS/f5_tts/model/backbones/dit.py @@ -15,7 +15,7 @@ from torch.utils.checkpoint import checkpoint from x_transformers.x_transformers import RotaryEmbedding -from GPT_SoVITS.f5_tts.model.modules import ( +from f5_tts.model.modules import ( TimestepEmbedding, ConvNeXtV2Block, ConvPositionEmbedding,