跳转到主要内容

概述

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

输入

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

输出

输出名称描述数据类型
模型已应用新计算数据类型的修改后模型MODEL
本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑

Source fingerprint (SHA-256): bc65f1e452d0122ad175a8b95f38a36503253c9908157037c516496e65c828e6