Stable Diffusion WebUI DirectML 模型加载失败问题分析与解决方案

问题现象描述

在使用 Stable Diffusion WebUI DirectML 版本时，部分用户遇到了模型加载失败的问题。具体表现为控制台日志中出现"failed to create model quickly; will retry using slow method"错误提示，随后系统变得异常缓慢，甚至需要强制关机。

错误原因分析

根据错误日志显示，问题主要源于以下几个方面：

1. Hugging Face 仓库访问失败：系统尝试从 Hugging Face 下载模型配置文件时返回401未授权错误，表明身份验证存在问题。

2. 依赖版本不兼容：transformers 和 diffusers 库的版本不匹配可能导致模型加载失败。

3. 磁盘空间不足：部分用户在首次运行时因磁盘空间不足导致下载中断，后续尝试时出现问题。

详细解决方案

方法一：调整依赖版本

1. 修改项目目录下的requirements.txt文件：

   • 将diffusers版本指定为0.29.2
   • 将transformers版本指定为4.30.2

2. 同样修改requirements_versions.txt文件中的对应版本号。

3. 删除现有的虚拟环境目录(venv)，然后重新运行启动脚本，让系统重新安装指定版本的依赖。

方法二：清理缓存并重新尝试

1. 删除Hugging Face缓存目录(通常位于用户目录下的.cache/huggingface)。

2. 确保有足够的磁盘空间(建议至少保留15GB可用空间)。

3. 重新运行WebUI，系统会自动重新下载所需文件。

方法三：检查模型文件完整性

1. 验证模型文件是否完整，特别是.safetensors文件。

2. 确保模型文件放置在正确的目录下(models/Stable-diffusion/)。

3. 尝试使用其他模型文件测试是否会出现相同问题。

技术背景说明

Stable Diffusion WebUI DirectML 是专为AMD显卡优化的版本，它使用DirectML作为后端来加速计算。在模型加载过程中，系统会尝试两种方式：

1. 快速方法：直接从内存或缓存加载预编译的模型组件。

2. 慢速方法：当快速方法失败时，系统会尝试从原始配置文件重建模型结构，这需要下载额外的组件并消耗更多资源。

预防措施建议

1. 保持足够的磁盘空间：建议在系统盘保留至少20GB可用空间，在模型存储盘保留更多空间。

2. 定期清理缓存：Hugging Face的缓存可能会占用大量空间，定期清理可以避免问题。

3. 使用稳定的依赖版本：不要随意升级核心依赖库，除非确定新版本已解决兼容性问题。

4. 监控系统资源：在首次运行或加载新模型时，注意观察系统资源使用情况，避免因资源耗尽导致系统无响应。

通过以上方法，大多数用户应该能够解决模型加载失败的问题。如果问题仍然存在，建议检查系统日志获取更详细的错误信息，或者考虑完全重新安装WebUI环境。