mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

dont use timeembedding

Browse Source
This commit is contained in:
mrq 2024-11-21 23:14:52 -06:00
parent 2cef97e43f
commit 8aafae91fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/models/base.py
View File

@ -537,7 +537,7 @@ class Base(nn.Module):
# experimental NAR-only mode # experimental NAR-only mode
self.len_emb = Embedding(11, d_model) 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": if attention_backend == "auto":
attention_backend = "sdpa" attention_backend = "sdpa"