Merge pull request #3149 from patrickvonplaten/fix_renaming_error

fix missed BartForMaskedLM renaming

Merge pull request #3149 from patrickvonplaten/fix_renaming_error
fix missed BartForMaskedLM renaming
c8035e11 · Patrick von Platen · GitHub · 857e0a0d · 58fc8f97 · c8035e11
Unverified Commit c8035e11 authored Mar 06, 2020 by Patrick von Platen Committed by GitHub Mar 06, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_bart.py tests/test_modeling_bart.py +1 -1

No files found.
--- a/tests/test_modeling_bart.py
+++ b/tests/test_modeling_bart.py
@@ -300,7 +300,7 @@ class BartHeadTests(unittest.TestCase):
    def test_generate_fp16(self):
        config, input_ids, batch_size = self._get_config_and_data(output_past=True)
        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)