Upload modeling_bert.py
Browse files- modeling_bert.py +2 -4
modeling_bert.py
CHANGED
|
@@ -186,8 +186,7 @@ def load_tf_weights_in_bert(model, config, tf_checkpoint_path):
|
|
| 186 |
array = np.transpose(array)
|
| 187 |
try:
|
| 188 |
if pointer.shape != array.shape:
|
| 189 |
-
raise ValueError(f"Pointer shape {pointer.shape} and array shape {
|
| 190 |
-
array.shape} mismatched")
|
| 191 |
except ValueError as e:
|
| 192 |
e.args += (pointer.shape, array.shape)
|
| 193 |
raise
|
|
@@ -845,8 +844,7 @@ class BertLayer(nn.Module):
|
|
| 845 |
if self.is_decoder and encoder_hidden_states is not None:
|
| 846 |
if not hasattr(self, "crossattention"):
|
| 847 |
raise ValueError(
|
| 848 |
-
f"If `encoder_hidden_states` are passed, {
|
| 849 |
-
self} has to be instantiated with cross-attention layers"
|
| 850 |
" by setting `config.add_cross_attention=True`"
|
| 851 |
)
|
| 852 |
|
|
|
|
| 186 |
array = np.transpose(array)
|
| 187 |
try:
|
| 188 |
if pointer.shape != array.shape:
|
| 189 |
+
raise ValueError(f"Pointer shape {pointer.shape} and array shape {array.shape} mismatched")
|
|
|
|
| 190 |
except ValueError as e:
|
| 191 |
e.args += (pointer.shape, array.shape)
|
| 192 |
raise
|
|
|
|
| 844 |
if self.is_decoder and encoder_hidden_states is not None:
|
| 845 |
if not hasattr(self, "crossattention"):
|
| 846 |
raise ValueError(
|
| 847 |
+
f"If `encoder_hidden_states` are passed, {self} has to be instantiated with cross-attention layers"
|
|
|
|
| 848 |
" by setting `config.add_cross_attention=True`"
|
| 849 |
)
|
| 850 |
|