Update models.py

NewKcriss · Oct 4, 2023 · 51063f3 · 51063f3
1 parent af80be6
commit 51063f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models.py b/models.py
@@ -16,7 +16,7 @@
 from Utils.JDC.model import JDCNet
 
 from Modules.diffusion.sampler import KDiffusion, LogNormalDistribution
-from Modules.diffusion.modules import Transformer1d
+from Modules.diffusion.modules import Transformer1d, StyleTransformer1d
 from Modules.diffusion.diffusion import AudioDiffusionConditional
 
 from Modules.discriminators import MultiPeriodDiscriminator, MultiResSpecDiscriminator, WavLMDiscriminator