Christopher H. commited on
Commit
eee1db3
·
verified ·
1 Parent(s): 4036053

print shape of used embeddings

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -1163,6 +1163,7 @@ class WhisperEncoder(WhisperPreTrainedModel):
1163
 
1164
  # CUSTOM
1165
  position_embeddings = self.get_position_embeddings(sequence_length)
 
1166
 
1167
 
1168
  hidden_states = inputs_embeds + position_embeddings
 
1163
 
1164
  # CUSTOM
1165
  position_embeddings = self.get_position_embeddings(sequence_length)
1166
+ print(f"Position_embeddings shape: {position_embeddings.shape}")
1167
 
1168
 
1169
  hidden_states = inputs_embeds + position_embeddings