Update README.md

Fix link to checkpoint repo