Fix checkpoint keys: strip torch.compile _orig_mod prefixes dda162f verified bryanlincoln commited on May 29