Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e7cfc1a313cc928e962bb8699868f5dcf46f11eb"
Commit 2708b44e authored by Lysandre's avatar Lysandre
Browse files

Patch ALBERT with heads in TensorFlow

parent 1abd53b1
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
src/transformers/modeling_tf_albert.py
View file @ 2708b44e
...@@ -506,14 +506,14 @@ class TFAlbertMainLayer(tf.keras.layers.Layer): ...@@ -506,14 +506,14 @@ class TFAlbertMainLayer(tf.keras.layers.Layer):
raise NotImplementedError raise NotImplementedError
def call( def call(
self, self,
inputs, inputs,
attention_mask=None, attention_mask=None,
token_type_ids=None, token_type_ids=None,
position_ids=None, position_ids=None,
head_mask=None, head_mask=None,
inputs_embeds=None, inputs_embeds=None,
training=False, training=False,
): ):
if isinstance(inputs, (tuple, list)): if isinstance(inputs, (tuple, list)):
input_ids = inputs[0] input_ids = inputs[0]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment