Fix ESM checkpoints for tests (#20436)

Commit

3 years ago

Fix ESM checkpoints for tests (#20436) * Re-enable TF ESM tests, make sure we use facebook checkpoints * make fixup