diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 50ae8dc30c..7373fb1567 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -9,7 +9,6 @@ on: push: branches: - ci_* - - framework-agnostic-tokenizers repository_dispatch: schedule: - cron: "0 0 * * *" diff --git a/tests/test_modeling_bart.py b/tests/test_modeling_bart.py index 76ebe2d3d9..2cadaa09c2 100644 --- a/tests/test_modeling_bart.py +++ b/tests/test_modeling_bart.py @@ -492,7 +492,9 @@ class BartModelIntegrationTests(unittest.TestCase): inputs_dict = prepare_bart_inputs_dict(model.config, input_ids) # Test that model hasn't changed with torch.no_grad(): - batched_logits, features = model(**inputs_dict) + outputs = model(**inputs_dict) + + batched_logits = outputs[0] expected_shape = torch.Size((2, 3)) self.assertEqual(batched_logits.shape, expected_shape) expected_slice = torch.Tensor([[0.1907, 1.4342, -1.0289]]).to(torch_device) diff --git a/tests/test_modeling_mbart.py b/tests/test_modeling_mbart.py index c394e62f04..6a1afd9d76 100644 --- a/tests/test_modeling_mbart.py +++ b/tests/test_modeling_mbart.py @@ -62,7 +62,7 @@ class AbstractSeq2SeqIntegrationTest(unittest.TestCase): @classmethod def setUpClass(cls): - cls.tokenizer = AutoTokenizer.from_pretrained(cls.checkpoint_name) + cls.tokenizer = AutoTokenizer.from_pretrained(cls.checkpoint_name, use_fast=False) return cls @cached_property