解决ebook2audiobook项目中XTTS模型加载失败问题

问题背景

在使用ebook2audiobook项目将电子书转换为有声书时，用户遇到了一个关于XTTS模型加载失败的错误。错误信息显示WeightsUnpickler error: Unsupported class TTS.tts.configs.xtts_config.XttsConfig，这表明系统无法正确加载XTTS模型的配置文件。

错误分析

该错误通常发生在以下情况：

1. 模型配置文件与当前安装的TTS库版本不兼容
2. PyTorch版本与TTS库版本不匹配
3. 缓存中存在旧版本的依赖项导致冲突

解决方案

1. 清理环境并重新安装

首先需要彻底清理现有的Python环境：

# 删除conda环境
rm -rf python_env
rm -rf ~/miniconda3

# 清理pip缓存
pip cache purge

2. 确保使用正确的架构版本

特别是在Apple Silicon(M1/M2)或ARM64架构的设备上，需要确保安装的是arm64版本的miniconda，而不是x86_64版本。

3. 更新关键依赖项

在重新创建环境后，应确保以下关键依赖项是最新兼容版本：

# 激活conda环境
conda activate ./python_env

# 更新PyTorch和TTS库
pip install --upgrade torch coqui-tts

4. 验证版本

安装完成后，验证关键组件的版本：

pip show torch coqui-tts

确保torch版本至少为2.1.0，推荐使用2.5.1版本。

技术细节

这个问题的根本原因是模型权重文件与当前运行时环境不兼容。当PyTorch尝试以weights_only=True模式加载模型时，会进行安全检查，防止潜在的恶意代码执行。如果模型配置文件类不在PyTorch的白名单中，就会抛出这个错误。

预防措施

为了避免类似问题，建议：

1. 在开始新项目前总是清理pip缓存
2. 使用虚拟环境隔离不同项目的依赖
3. 定期更新关键依赖项
4. 对于ARM架构设备，特别注意安装对应架构的软件包

总结

通过彻底清理环境、确保架构兼容性以及更新关键依赖项，可以成功解决XTTS模型加载失败的问题。这个问题提醒我们在处理AI模型和大型依赖项时，环境一致性和版本管理的重要性。

对于ebook2audiobook项目的用户，特别是在Apple Silicon设备上使用时，遵循上述步骤可以确保TTS功能的正常工作，顺利将电子书转换为有声书。