Method model

tests/test_modeling_bart.py:230–235

Only load the model if needed.

(self)

    @cached_property
    def model(self):
        """Only load the model if needed."""
        model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-en-ro").to(torch_device)
        if "cuda" in torch_device:
            model = model.half()
        return model

    @slow
    @unittest.skip("This has been failing since June 20th at least.")