Modeling fixed

modeling_nicheformer.py

@@ -116,8 +116,6 @@ class NicheformerModel(NicheformerPreTrainedModel):
         # Get token embeddings and positional encodings
         token_embedding = self.embeddings(input_ids)
         
-        print("token_embedding NicheformerModel: ", token_embedding)
-
         if self.config.learnable_pe:
             pos_embedding = self.positional_embedding(self.pos.to(token_embedding.device))
             embeddings = self.dropout(token_embedding + pos_embedding)