SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix issue: #1962, input's shape seem to cause error in 2.2.0 version tf_albert_model

Browse Source
This commit is contained in:
root
2019-11-27 09:58:38 +00:00
committed by Lysandre Debut
parent 45d767297a
commit 88b317739f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformers/modeling_tf_albert.py
View File

@@ -645,7 +645,7 @@ class TFAlbertModel(TFAlbertPreTrainedModel):
if input_ids is not None and inputs_embeds is not None: if input_ids is not None and inputs_embeds is not None:
raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time") raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
elif input_ids is not None: elif input_ids is not None:
input_shape = input_ids.shape input_shape = tf.shape(input_ids)
elif inputs_embeds is not None: elif inputs_embeds is not None:
input_shape = inputs_embeds.shape[:-1] input_shape = inputs_embeds.shape[:-1]
else: else: