update

toolbox/torchaudio/models/nx_clean_unet/modeling_nx_clean_unet.py

@@ -172,9 +172,9 @@ class NXCleanUNet(nn.Module):
             attention_heads=config.tsfm_attention_heads,
             num_blocks=config.tsfm_num_blocks,
             dropout_rate=config.tsfm_dropout_rate,
-            chunk_size=config.chunk_size,
-            num_left_chunks=config.num_left_chunks,
-            num_right_chunks=config.num_right_chunks,
+            chunk_size=config.tsfm_chunk_size,
+            num_left_chunks=config.tsfm_num_left_chunks,
+            num_right_chunks=config.tsfm_num_right_chunks,
         )
         self.up_sampling = UpSampling(
             num_layers=config.down_sampling_num_layers,