SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
get correct checkpoint
This commit is contained in:
Arthur
2022-12-23 19:55:17 +01:00
committed by GitHub
parent cab7799f7b
commit a081f292ca
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/roberta_prelayernorm/test_modeling_roberta_prelayernorm.py
View File

@@ -524,7 +524,7 @@ class RobertaPreLayerNormModelIntegrationTest(TestCasePlus):
@slow @slow
def test_inference_no_head(self): 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]]) input_ids = torch.tensor([[0, 31414, 232, 328, 740, 1140, 12695, 69, 46078, 1588, 2]])
with torch.no_grad(): with torch.no_grad():