You probably saved a dict {"model": models_sd, ...} as checkpoint, so load checkpoint["model"]. (The key information here is not the keys it is missing but the ones that are unexpected.)
Best regards
Thomas