Update chatNT.py

chatNT.py

@@ -416,9 +416,11 @@ class TorchBioBrainDecoder(nn.Module):
                 _,
                 bio_embed_dim,
             ) = projected_bio_embeddings.shape
-
+            
             # Insert the bio embeddings at the SEQ token positions
             processed_tokens_ids = english_token_ids.clone()
+            print("(debug) Inside : processed tokens ids shape : ", processed_tokens_ids.shape)
+            print("(debug) Inside : projected bio embeddings shape : ", projected_bio_embeddings.shape)
             for bio_seq_num in range(num_bio_sequences):
                 tokens_embeddings, processed_tokens_ids = self.insert_embeddings(
                     processed_tokens_ids,
@@ -426,6 +428,7 @@ class TorchBioBrainDecoder(nn.Module):
                     projected_bio_embeddings[:, bio_seq_num, :, :],
                     bio_seq_num=bio_seq_num,
                 )
+                print("After call : ", tokens_embeddings.shape)
 
         # Regular GPT pass through
         print("(debug) tokens embeddings shape : ", tokens_embeddings.shape)