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

livid

Browse Source
This commit is contained in:
mrq 2024-12-19 19:25:27 -06:00
parent 8838babcba
commit e7e7f48043
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/models/base.py
View File

@ -546,6 +546,8 @@ class Base(nn.Module):
if self.version >= 3: if self.version >= 3:
self.langs_emb = Embedding(n_langs, d_model) if n_langs > 0 else None self.langs_emb = Embedding(n_langs, d_model) if n_langs > 0 else None
self.tasks_emb = Embedding(n_tasks, d_model) if n_tasks > 0 else None self.tasks_emb = Embedding(n_tasks, d_model) if n_tasks > 0 else None
self.capabilities += ["lang"]
# never actually got added... I kept forgetting to classify all my audio for speaker's tone # never actually got added... I kept forgetting to classify all my audio for speaker's tone
if self.version >= 4: if self.version >= 4:
self.tones_emb = Embedding(n_tones, d_model) if n_tones > 0 else None self.tones_emb = Embedding(n_tones, d_model) if n_tones > 0 else None