Update chatNT.py

chatNT.py

@@ -1759,7 +1759,7 @@ class TorchMultiModalPerceiverResampler(nn.Module):
         """
         Create the blocks of attention layers and applies them.
         """
-        for layer in self.layers:
+        for layer_idx, layer in enumerate(self.layers):
             concat_input_1 = torch.cat([xf_1, x], dim=1)
             concat_input_2 = torch.cat([xf_2, x], dim=1)