Skip to content

GitLab

Unverified Commit 95e00d08 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Clean special token init in modeling_....py (#3264) 

* make style

* fix conflicts
parent 8becb732
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_modeling_tf_transfo_xl.py
View file @ 95e00d08
......@@ -107,7 +107,7 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase):
d_inner=self.d_inner,
div_val=self.div_val,
n_layer=self.num_hidden_layers,
eos_token_ids=self.eos_token_id,
eos_token_id=self.eos_token_id,
)
return (config, input_ids_1, input_ids_2, lm_labels)
......
tests/test_modeling_transfo_xl.py
View file @ 95e00d08
......@@ -103,7 +103,7 @@ class TransfoXLModelTest(ModelTesterMixin, unittest.TestCase):
d_inner=self.d_inner,
div_val=self.div_val,
n_layer=self.num_hidden_layers,
eos_token_ids=self.eos_token_id,
eos_token_id=self.eos_token_id,
)
return (config, input_ids_1, input_ids_2, lm_labels)
......
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