KModel 通过 strict=False 方式默默加载不匹配的检查点

摘要

KModel 在默认情况下使用严格的 strict=False 回退加载检查点状态字典。以较新的 torch.nn.utils.parametrizations.weight_norm 格式保存的微调后的检查点使用了诸如 *.parametrizations.weight.original0/*.parametrizations.weight.original1 等键，而不是模型模块所期望的 *.weight_g/*.weight_v 键。由于加载不严格，此类检查点虽然“加载成功”，但每个权重归一化层都处于随机初始化状态。结果是纯噪声，而不是语音，没有错误和警告。这影响了至少一个公开可用的非英语语言微调族群（例如“kikiri-tts”检查点）。