GPT-SoVITS项目在PyTorch 2.6+版本中的模型加载问题解析

问题背景

在GPT-SoVITS语音合成项目中，当用户使用PyTorch 2.6及以上版本时，可能会遇到模型加载失败的问题。这个问题主要出现在加载SoVITS模型和GPT模型时，系统会抛出关于weights_only参数的异常。

技术原理分析

PyTorch 2.6版本对模型加载机制进行了重要安全更新。默认情况下，torch.load()函数的weights_only参数从False变更为True。这一变更旨在提高模型加载的安全性，防止潜在的恶意代码执行。

当weights_only=True时，PyTorch会严格限制可以反序列化的对象类型，只允许加载包含基础数据类型和张量的模型文件。如果模型文件中包含自定义类（如项目中的HParams类），加载就会失败并抛出异常。

具体解决方案

针对GPT-SoVITS项目，我们有以下两种解决方案：

1. 降低PyTorch版本： 将PyTorch降级到2.6之前的版本（如2.5.x），这是最直接的解决方法，因为旧版本默认weights_only=False。

2. 修改代码参数： 对于SoVITS模型加载，可以修改代码中的torch.load()调用，显式设置weights_only=False：

   dict_s2 = torch.load(sovits_path, map_location="cpu", weights_only=False)
   

   需要注意的是，GPT模型的加载不需要修改，保持原样即可。

安全注意事项

虽然设置weights_only=False可以解决问题，但用户需要注意：

• 只从可信来源加载模型文件
• 了解潜在的安全风险（可能执行任意代码）
• 在生产环境中，建议优先考虑降低PyTorch版本而非修改参数

最佳实践建议

对于GPT-SoVITS项目的用户，我们建议：

1. 如果环境允许，优先使用PyTorch 2.5.x版本
2. 如果必须使用PyTorch 2.6+，确保模型文件来源可靠后再使用weights_only=False
3. 关注项目官方更新，未来版本可能会提供更完善的兼容性解决方案

通过以上分析和解决方案，用户应该能够顺利地在不同PyTorch版本下运行GPT-SoVITS项目。