sgugger's picture
Fix checkpoint
dd565b6