Skip to content

GitLab

Commit 556442af authored by thomwolf's avatar thomwolf
Browse files

hot fix

parent 160b5d60
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_transformers/modeling_tf_xlm.py
View file @ 556442af
...@@ -53,7 +53,7 @@ def load_xlm_pt_weights_in_tf2(tf_model, pytorch_checkpoint_path): ...@@ -53,7 +53,7 @@ def load_xlm_pt_weights_in_tf2(tf_model, pytorch_checkpoint_path):
langs_list = tf.constant([[1, 1, 0, 0, 1], [1, 1, 1, 0, 0], [1, 0, 0, 1, 1]]) langs_list = tf.constant([[1, 1, 0, 0, 1], [1, 1, 1, 0, 0], [1, 0, 0, 1, 1]])
else: else:
langs_list = None langs_list = None
tf_inputs = [inputs_list, attns_list), langs_list] tf_inputs = [inputs_list, attns_list, langs_list]
tfo = tf_model(tf_inputs, training=False) tfo = tf_model(tf_inputs, training=False)
return load_pytorch_checkpoint_in_tf2_model(tf_model, pytorch_checkpoint_path, tf_inputs=tf_inputs) return load_pytorch_checkpoint_in_tf2_model(tf_model, pytorch_checkpoint_path, tf_inputs=tf_inputs)
......
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