Update chatNT.py

chatNT.py

@@ -640,7 +640,7 @@ class TorchMultiOmicsModel(PreTrainedModel):
 
     def forward(
         self,
-        multi_omics_tokens_ids: tuple[torch.Tensor, torch.Tensor],
+        multi_omics_tokens_ids: tuple[torch.Tensor, torch.Tensor | None],
         projection_english_tokens_ids: torch.Tensor,
         projected_bio_embeddings: torch.Tensor = None,
     ) -> dict[str, torch.Tensor]:
@@ -671,8 +671,9 @@ class TorchMultiOmicsModel(PreTrainedModel):
         """
         english_token_ids, bio_token_ids = multi_omics_tokens_ids
         english_token_ids = english_token_ids.clone()
-        bio_token_ids = bio_token_ids.clone()
         projection_english_tokens_ids = projection_english_tokens_ids.clone()
+        if bio_token_ids is not None:
+            bio_token_ids = bio_token_ids.clone()
         if projected_bio_embeddings is not None:
             projected_bio_embeddings = projected_bio_embeddings.clone()