From 197d51718192d6a59c5e0b1f7358817ad3287d1f Mon Sep 17 00:00:00 2001 From: mrq Date: Mon, 27 May 2024 17:09:35 -0500 Subject: [PATCH] ugh --- vall_e/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/models/base.py b/vall_e/models/base.py index cd1dc69..3e808b4 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -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 )