[RobertaPreLayernom] Fixes the CI daily test (#20886)

get correct checkpoint
2022-12-23 19:55:17 +01:00
parent cab7799f7b
commit a081f292ca
1 changed files with 1 additions and 1 deletions
--- a/tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py
+++ b/tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py
@@ -524,7 +524,7 @@ class RobertaPreLayerNormModelIntegrationTest(TestCasePlus):

    @slow
    def test_inference_no_head(self):
-        model = RobertaPreLayerNormModel.from_pretrained("princeton-nlp/efficient_mlm_m0.40")
+        model = RobertaPreLayerNormModel.from_pretrained("andreasmadsen/efficient_mlm_m0.40")

        input_ids = torch.tensor([[0, 31414, 232, 328, 740, 1140, 12695, 69, 46078, 1588, 2]])
        with torch.no_grad():