本文档由 AI 生成。如果您发现任何错误或有改进建议,欢迎贡献! 在 GitHub 上编辑
概述
ModelComputeDtype 节点用于更改模型在处理过程中的计算数据类型(精度)。它会创建输入模型的副本,并应用所选的精度设置,这有助于根据硬件情况优化内存使用和性能。该节点对于调试和测试不同精度配置非常有用。输入
| 参数 | 数据类型 | 是否必填 | 取值范围 | 描述 |
|---|---|---|---|---|
模型 | MODEL | 是 | - | 要修改计算数据类型的输入模型 |
dtype | STRING | 是 | ”default" "fp32" "fp16" "bf16” | 应用于模型的计算数据类型(默认值:“default”) |
输出
| 输出名称 | 数据类型 | 描述 |
|---|---|---|
模型 | MODEL | 已应用新计算数据类型的修改后模型 |
Source fingerprint (SHA-256):
bc65f1e452d0122ad175a8b95f38a36503253c9908157037c516496e65c828e6