Spaces:
Build error
Build error
PeteBleackley
commited on
Commit
·
58d8758
1
Parent(s):
30efe84
Needed more arguments
Browse files
qarac/models/QaracDecoderModel.py
CHANGED
|
@@ -77,11 +77,17 @@ class QaracDecoderHead(keras.layers.Layer):
|
|
| 77 |
l0 = self.layer_0(vectors,
|
| 78 |
attentions,
|
| 79 |
None,
|
|
|
|
|
|
|
|
|
|
| 80 |
False,
|
| 81 |
training)
|
| 82 |
return self.head(self.layer1(l0.last_hidden_state[:,1:],
|
| 83 |
attention_mask,
|
| 84 |
None,
|
|
|
|
|
|
|
|
|
|
| 85 |
False,
|
| 86 |
training))
|
| 87 |
|
|
|
|
| 77 |
l0 = self.layer_0(vectors,
|
| 78 |
attentions,
|
| 79 |
None,
|
| 80 |
+
None,
|
| 81 |
+
None,
|
| 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 |
|