diff --git a/vall_e/models/base.py b/vall_e/models/base.py index d2fefbf..dd1ca93 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -537,7 +537,7 @@ class Base(nn.Module): # experimental NAR-only mode self.len_emb = Embedding(11, d_model) - self.time_emb = TimeEmbedding(d_model) # if not masking_ratio else None + self.time_emb = None # TimeEmbedding(d_model) # if not masking_ratio else None if attention_backend == "auto": attention_backend = "sdpa"