PeteBleackley commited on
Commit
50de02e
·
1 Parent(s): 58d8758

Fixed typo

Browse files
Files changed (1) hide show
  1. qarac/models/QaracDecoderModel.py +8 -8
qarac/models/QaracDecoderModel.py CHANGED
@@ -82,14 +82,14 @@ class QaracDecoderHead(keras.layers.Layer):
82
  None,
83
  False,
84
  training)
85
- return self.head(self.layer1(l0.last_hidden_state[:,1:],
86
- attention_mask,
87
- None,
88
- None,
89
- None,
90
- None,
91
- False,
92
- training))
93
 
94
  class QaracDecoderModel(transformers.TFPreTrainedModel,transformers.generation_tf_utils.TFGenerationMixin):
95
 
 
82
  None,
83
  False,
84
  training)
85
+ return self.head(self.layer_1(l0.last_hidden_state[:,1:],
86
+ attention_mask,
87
+ None,
88
+ None,
89
+ None,
90
+ None,
91
+ False,
92
+ training))
93
 
94
  class QaracDecoderModel(transformers.TFPreTrainedModel,transformers.generation_tf_utils.TFGenerationMixin):
95