Remove debugging
Browse files
model.py
CHANGED
|
@@ -1164,9 +1164,7 @@ class WhisperEncoder(WhisperPreTrainedModel):
|
|
| 1164 |
sequence_length = inputs_embeds.shape[1]
|
| 1165 |
|
| 1166 |
# CUSTOM
|
| 1167 |
-
position_embeddings = self.get_position_embeddings(sequence_length)
|
| 1168 |
-
print(f"Position_embeddings shape: {position_embeddings.shape}")
|
| 1169 |
-
|
| 1170 |
|
| 1171 |
hidden_states = inputs_embeds + position_embeddings
|
| 1172 |
hidden_states = nn.functional.dropout(hidden_states, p=self.dropout, training=self.training)
|
|
|
|
| 1164 |
sequence_length = inputs_embeds.shape[1]
|
| 1165 |
|
| 1166 |
# CUSTOM
|
| 1167 |
+
position_embeddings = self.get_position_embeddings(sequence_length)
|
|
|
|
|
|
|
| 1168 |
|
| 1169 |
hidden_states = inputs_embeds + position_embeddings
|
| 1170 |
hidden_states = nn.functional.dropout(hidden_states, p=self.dropout, training=self.training)
|