Fix ESM checkpoints for tests (#20436)

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

* make fixup

tests/models/esm/test_modeling_esmfold.py

@@ -247,7 +247,7 @@ class EsmFoldModelTest(ModelTesterMixin, unittest.TestCase):
 class EsmModelIntegrationTest(TestCasePlus):
     @slow
     def test_inference_protein_folding(self):
-        model = EsmForProteinFolding.from_pretrained("Rocketknight1/esmfold_v1").float()
+        model = EsmForProteinFolding.from_pretrained("facebook/esmfold_v1").float()
         model.eval()
         input_ids = torch.tensor([[0, 6, 4, 13, 5, 4, 16, 12, 11, 7, 2]])
         position_outputs = model(input_ids)["positions"]