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

Fix ESM checkpoints for tests (#20436)

Browse Source 
* Re-enable TF ESM tests, make sure we use facebook checkpoints

* make fixup
This commit is contained in:
Matt
2022-11-28 13:19:28 +00:00
committed by GitHub
parent f244a97801
commit 72b19ca680
3 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/esm/test_modeling_esm.py
View File

@@ -274,7 +274,7 @@ class EsmModelIntegrationTest(TestCasePlus):
@slow
def test_inference_masked_lm(self):
with torch.no_grad():
model = EsmForMaskedLM.from_pretrained("Rocketknight1/esm2_t6_8M_UR50D")
model = EsmForMaskedLM.from_pretrained("facebook/esm2_t6_8M_UR50D")
model.eval()
input_ids = torch.tensor([[0, 1, 2, 3, 4, 5]])
output = model(input_ids)[0]
@@ -292,7 +292,7 @@ class EsmModelIntegrationTest(TestCasePlus):
@slow
def test_inference_no_head(self):
with torch.no_grad():
model = EsmModel.from_pretrained("Rocketknight1/esm2_t6_8M_UR50D")
model = EsmModel.from_pretrained("facebook/esm2_t6_8M_UR50D")
model.eval()
input_ids = torch.tensor([[0, 6, 4, 13, 5, 4, 16, 12, 11, 7, 2]])