Update chatNT.py

chatNT.py

@@ -725,6 +725,11 @@ class TorchMultiOmicsModel(PreTrainedModel):
         )
 
         outs = {"logits": logits, "projected_bio_embeddings": projected_bio_embeddings}
+        
+        # Just for debugging
+        print("(debug) remember to remove bio_embeddings storage")
+        for i, embed in enumerate(bio_embeddings_list):
+            outs[f"bio_embeddings_list_{i}"] = embed
 
         return outs