From 4e7d885542de1771cc0a91ffd34c0ebde6eae500 Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 28 Feb 2025 18:06:41 -0600 Subject: [PATCH] lol --- vall_e/models/base_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/models/base_v2.py b/vall_e/models/base_v2.py index bb39396..9b2ae90 100644 --- a/vall_e/models/base_v2.py +++ b/vall_e/models/base_v2.py @@ -449,9 +449,9 @@ class Base_V2(nn.Module): hidden_act="gelu", is_encoder_decoder=False, is_decoder=True, - output_norm=not per_level_normalization, # moves the LN out to the decoder #gradient_checkpointing=self.gradient_checkpointing, ) + self.model_config.output_norm = not per_level_normalization # moves the LN out to the decoder self.model_config.attn_mode = attention_backend self.model = LlamaModel(self.model_config)