跳转到主要内容
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑

概述

ModelComputeDtype 节点用于更改模型在处理过程中的计算数据类型(精度)。它会创建输入模型的副本,并应用所选的精度设置,这有助于根据硬件情况优化内存使用和性能。该节点对于调试和测试不同精度配置非常有用。

输入

参数数据类型是否必填取值范围描述
模型MODEL-要修改计算数据类型的输入模型
dtypeSTRING”default"
"fp32"
"fp16"
"bf16”
应用于模型的计算数据类型(默认值:“default”)

输出

输出名称数据类型描述
模型MODEL已应用新计算数据类型的修改后模型

Source fingerprint (SHA-256): bc65f1e452d0122ad175a8b95f38a36503253c9908157037c516496e65c828e6