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

fix renaming problem

Browse Source
This commit is contained in:
patrickvonplaten
2020-03-06 00:35:47 +01:00
parent 857e0a0d3b
commit 58fc8f97a3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_bart.py
View File

@@ -300,7 +300,7 @@ class BartHeadTests(unittest.TestCase):
def test_generate_fp16(self): def test_generate_fp16(self):
config, input_ids, batch_size = self._get_config_and_data(output_past=True) config, input_ids, batch_size = self._get_config_and_data(output_past=True)
attention_mask = input_ids.ne(1) attention_mask = input_ids.ne(1)
lm_model = BartForMaskedLM(config).eval().to(torch_device).half() lm_model = BartForConditionalGeneration(config).eval().to(torch_device).half()
lm_model.generate(input_ids, attention_mask) lm_model.generate(input_ids, attention_mask)