riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-24 14:58:56 +06:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

fixed typo

Browse Source
This commit is contained in:
Patrick von Platen 2020-03-09 20:01:20 +01:00
parent a2c8e516c2
commit 374deef48d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_utils.py
View File

@ -855,7 +855,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin):
cur_len = 1
# put model in generation mode if it has one
if hasattr(self.model, "generation_mode"):
if hasattr(self.model, "decoder") and hasattr(self.model.decoder, "generation_mode"):
self.model.decoder.generation_mode = True
else:
encoder_inputs = None

2
tests/test_modeling_bart.py
View File

@ -287,7 +287,7 @@ class BartHeadTests(unittest.TestCase):
new_input_ids = lm_model.generate(
input_ids.clone(), num_return_sequences=1, num_beams=2, no_repeat_ngram_size=3, max_length=max_length
)
self.assertEqual(new_input_ids.shape, (input_ids.shape[0], max_length - 1))
self.assertEqual(new_input_ids.shape, (input_ids.shape[0], max_length))
# TODO(SS): uneven length batches, empty inputs
def test_shift_tokens_right(self):