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

get correct checkpoint

[RobertaPreLayernom] Fixes the CI daily test (#20886)
get correct checkpoint
a081f292 · Arthur · GitHub · cab7799f · a081f292
Unverified Commit a081f292 authored Dec 23, 2022 by Arthur Committed by GitHub Dec 23, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py ...oberta_prelayernorm/test_modeling_roberta_prelayernorm.py +1 -1

No files found.
--- 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():