Spaces:
Build error
Build error
PeteBleackley
commited on
Commit
·
50de02e
1
Parent(s):
58d8758
Fixed typo
Browse files
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.
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
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 |
|