Update chatNT.py

chatNT.py

@@ -1763,8 +1763,8 @@ class TorchMultiModalPerceiverResampler(nn.Module):
             concat_input_1 = torch.cat([xf_1, x], dim=1)
             concat_input_2 = torch.cat([xf_2, x], dim=1)
 
-            #outs[f"concat_input_1_{layer_idx}"] = concat_input_1.clone()
-            #outs[f"concat_input_2_{layer_idx}"] = concat_input_2.clone()
+            outs[f"PERCEIVER_RESAMPLER_concat_input_1_{layer_idx}"] = concat_input_1.clone()
+            outs[f"PERCEIVER_RESAMPLER_concat_input_2_{layer_idx}"] = concat_input_2.clone()
 
             output = layer(
                 x=x,
@@ -1774,7 +1774,7 @@ class TorchMultiModalPerceiverResampler(nn.Module):
                 attention_mask_2=attention_mask_2,
             )
             x = output["embeddings"]
-            #outs[f"attention_embeddings_{layer_idx}"] = output["embeddings"].clone()
+            outs[f"PERCEIVER_RESAMPLER_attention_embeddings_{layer_idx}"] = output["embeddings"].clone()
 
         return x, outs