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 |
|