本文档由 AI 生成。如果您发现任何错误或有改进建议，欢迎贡献！ 在 GitHub 上编辑ModelComputeDtype 节点允许您在推理过程中更改模型使用的计算数据类型。它会创建输入模型的副本并应用指定的数据类型设置，这有助于根据硬件能力优化内存使用和性能。该节点在调试和测试不同精度设置时特别有用。
输入参数
| 参数 | 数据类型 | 必填 | 取值范围 | 描述 |
|---|---|---|---|---|
模型 | MODEL | 是 | - | 需要应用新计算数据类型的输入模型 |
dtype | STRING | 是 | ”default" "fp32" "fp16" "bf16” | 要应用于模型的计算数据类型 |
输出结果
| 输出名称 | 数据类型 | 描述 |
|---|---|---|
模型 | MODEL | 已应用新计算数据类型的修改后模型 |