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

ugh

Browse Source
This commit is contained in:
mrq 2024-05-27 17:09:35 -05:00
parent 5af6f41c94
commit 197d517181
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

@ -907,7 +907,7 @@ class Base(nn.Module):
if name in ["text" ]:
text_batch.append( input )
elif name == "prom" and (quant_level is None or quant_level == 0 or not self.config.audio_embedding_sums):
elif name == "prom": # and (quant_level is None or quant_level == 0) and not self.config.audio_embedding_sums:
prom_batch.append( input[:, quant_level] if quant_level is not None else input )
elif name == "targ":
resp_batch.append( input )