InstaDeepAI
/

nucleotide-transformer-v2-50m-multi-species

Model card Files Files and versions

hdallatorre commited on Sep 19, 2023

Commit

ccc60d1

·

1 Parent(s): 50add8c

Update modeling_esm.py

Files changed (1) hide show

modeling_esm.py +0 -1

modeling_esm.py CHANGED Viewed

@@ -551,7 +551,6 @@ class EsmIntermediate(nn.Module):
         self.activation_fn = SiLU()
     def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        print("Forwarding in the custom ESM Intermediate!")
         hidden_states = self.dense(hidden_states)
         # GLU

         self.activation_fn = SiLU()
     def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
         hidden_states = self.dense(hidden_states)
         # GLU