ComfyUI-layerdiffuse项目中SD1.5模型采样时的sigmas键错误分析

在ComfyUI-layerdiffuse项目使用过程中，部分用户反馈在使用SD1.5模型配合Layer Diffuse Apply节点进行采样时，会遇到KeyError: 'sigmas'的错误。本文将深入分析这一问题的成因及可能的解决方案。

问题现象

当用户尝试运行包含Layer Diffuse Apply节点的SD1.5工作流时，KSampler阶段会抛出KeyError: 'sigmas'异常。值得注意的是，同样的工作流在切换为SDXL模型时则能正常运行。错误通常出现在注意力共享机制的实现代码中，具体是在尝试访问transformer_options字典中的'sigmas'键时失败。

技术背景

ComfyUI-layerdiffuse项目通过注意力共享机制优化了模型的推理过程。在SD1.5模型中，这一机制会尝试访问transformer_options字典中的sigmas参数，用于控制扩散过程中的噪声调度。然而在某些配置下，这个参数可能未被正确初始化或传递。

可能的原因分析

1. 模型配置差异：SDXL模型的配置默认不使用注意力共享机制，因此不会触发相关代码路径，这也是为什么SDXL模型不受影响的原因。

2. ComfyUI版本兼容性：不同版本的ComfyUI可能在参数传递机制上有所变化，导致某些必要参数未被正确传递。

3. xformers干扰：有案例表明，xformers的启用可能与某些采样过程产生冲突，导致参数传递异常。

解决方案建议

1. 检查ComfyUI版本：确保使用最新稳定版本的ComfyUI框架，避免因版本差异导致的参数传递问题。

2. 禁用xformers：在部分硬件配置下，xformers可能干扰正常的采样过程。可以尝试临时禁用xformers来验证问题是否解决。

3. 调试注意力共享：通过在attention_sharing.py中添加调试输出，可以确认问题发生的具体位置和上下文环境。

4. 参数初始化检查：确保在调用采样过程前，所有必要的参数（包括sigmas）都被正确初始化和传递。

最佳实践

对于遇到类似问题的用户，建议按照以下步骤排查：

1. 首先确认问题是否特定于SD1.5模型
2. 检查ComfyUI版本是否为最新
3. 尝试禁用xformers等可能产生干扰的优化模块
4. 在最小化的工作流中复现问题，排除其他节点干扰
5. 如有必要，添加调试输出以获取更多上下文信息

通过系统性的排查，大多数情况下可以定位到问题的根源并找到合适的解决方案。对于持续出现的问题，建议收集完整的错误日志和环境信息以便进一步分析。