Update chatNT.py

chatNT.py

@@ -733,6 +733,7 @@ class TorchMultiOmicsModel(PreTrainedModel):
         )
 
         outs["logits"] = logits
+        outs["projected_bio_embeddings_end"] = projected_bio_embeddings.clone()
 
         return outs