hdallatorre
commited on
Commit
•
ccc60d1
1
Parent(s):
50add8c
Update modeling_esm.py
Browse files- modeling_esm.py +0 -1
modeling_esm.py
CHANGED
@@ -551,7 +551,6 @@ class EsmIntermediate(nn.Module):
|
|
551 |
self.activation_fn = SiLU()
|
552 |
|
553 |
def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
|
554 |
-
print("Forwarding in the custom ESM Intermediate!")
|
555 |
hidden_states = self.dense(hidden_states)
|
556 |
|
557 |
# GLU
|
|
|
551 |
self.activation_fn = SiLU()
|
552 |
|
553 |
def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
|
|
|
554 |
hidden_states = self.dense(hidden_states)
|
555 |
|
556 |
# GLU
|