InstaDeepAI
/

ChatNT

Text Generation

feature-extraction

Model card Files Files and versions Community

Yanisadel commited on Apr 1

Commit

bdfd38c

·

1 Parent(s): c11bede

Update chatNT.py

Files changed (1) hide show

chatNT.py +1 -1

chatNT.py CHANGED Viewed

@@ -698,7 +698,7 @@ class TorchMultiOmicsModel(PreTrainedModel):
             if projected_bio_embeddings is None:
                 # Compute bio sequences embeddings
-                print("(debug) shape bio tokens ids : ", bio_tokens_ids.shape)
                 bio_embeddings_list = [
                     self.biobrain_encoder(bio_token_ids=bio_token_ids[:, bio_seq_num])
                     for bio_seq_num in range(num_bio_sequences)

             if projected_bio_embeddings is None:
                 # Compute bio sequences embeddings
+                print("(debug) shape bio tokens ids : ", bio_token_ids.shape)
                 bio_embeddings_list = [
                     self.biobrain_encoder(bio_token_ids=bio_token_ids[:, bio_seq_num])
                     for bio_seq_num in range(num_bio_sequences)