From 0280e7225794042a0ab072312413ec2431326f9e Mon Sep 17 00:00:00 2001 From: mrq Date: Mon, 17 Mar 2025 21:49:45 -0500 Subject: [PATCH] ugh --- 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 bb12eb7..5cd90fe 100644 --- a/vall_e/models/base_v2.py +++ b/vall_e/models/base_v2.py @@ -1113,7 +1113,7 @@ class Base_V2(nn.Module): is_causal = [ l in causal_levels for l in classifier_levels ] if self.noncausal_masks else [ True for l in classifier_levels ] if self.parallel_attention_mask_dropout > 0: - is_causal = [ True if random.random() < parallel_attention_mask_dropout else m for m in is_causal ] + is_causal = [ True if random.random() < self.parallel_attention_mask_dropout else m for m in is_causal ] # create special masks # to-do, create it if mixed (although I expect this model to be purely non-causal)