We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent db8e21d commit b83d1dbCopy full SHA for b83d1db
1 file changed
nemo/collections/tts/models/magpietts.py
@@ -4210,9 +4210,9 @@ def _initialize_chunked_attn_prior(
4210
prior_weights = self.chunked_inference_config.prior_weights_init
4211
_attn_prior[_idx, :, :current_starting_point] = prior_epsilon * prior_epsilon
4212
for offset, weight in enumerate(prior_weights[:5]):
4213
- idx = current_starting_point + offset
4214
- if idx < max_text_len:
4215
- _attn_prior[_idx, :, idx] = weight
+ current_offset_idx = current_starting_point + offset
+ if current_offset_idx < max_text_len:
+ _attn_prior[_idx, :, current_offset_idx] = weight
4216
4217
return _attn_prior
4218
0 commit comments