Update chatNT.py

chatNT.py

@@ -728,8 +728,9 @@ class TorchMultiOmicsModel(PreTrainedModel):
         
         # 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
+        if projected_bio_embeddings is not None:
+            for i, embed in enumerate(bio_embeddings_list):
+                outs[f"bio_embeddings_list_{i}"] = embed
 
         return outs