OpenNLPLab commited on
Commit
a29ea53
1 Parent(s): ec5dbd2

Update modeling_transnormer.py

Browse files
Files changed (1) hide show
  1. modeling_transnormer.py +0 -3
modeling_transnormer.py CHANGED
@@ -749,9 +749,6 @@ class TransnormerModel(TransnormerPreTrainedModel):
749
 
750
  if output_attentions:
751
  all_self_attns += (layer_outputs[1],)
752
-
753
- # if idx == 0:
754
- # break
755
 
756
  hidden_states = self.final_norm(hidden_states)
757
 
 
749
 
750
  if output_attentions:
751
  all_self_attns += (layer_outputs[1],)
 
 
 
752
 
753
  hidden_states = self.final_norm(hidden_states)
754