选择模型设备 - ComfyUI Built-in Node Documentation

Select Model Device 节点允许你手动选择扩散模型在哪个设备（CPU 或特定 GPU）上运行。

Select Model Device 节点允许你手动选择扩散模型在哪个设备（CPU 或特定 GPU）上运行。它可以将模型移动到其他设备，并自动处理与其他多 GPU 节点的冲突。选择 "default" 会恢复模型加载器最初选择的设备。

输入

参数详情：

• "default"：恢复模型加载器分配的加载设备和卸载设备，即使之前调用过 Select Model Device 也是如此。
• "cpu"：将加载设备和卸载设备都固定到 CPU。
• "gpu:N"：将加载设备固定到第 N 个可用 GPU（例如，第一个 GPU 使用 "gpu:0"）。卸载设备会恢复为加载器最初的选择。

重要说明：

• 如果当前机器上不存在所请求的设备（例如，在 2-GPU 机器上创建的工作流在 1-GPU 机器上打开），该节点会原样传递模型并记录一条消息，而不是报错。输入验证期间允许未知的 gpu:N 值，因此可移植工作流不会过早报错。
• 如果模型已经在所请求的设备上，该节点会使用快速路径，不会重新加载模型。
• 当所请求的设备与输入模型当前所在的设备不同时，会使用加载器的重新加载工厂生成一个新模型，以便新的 patcher 在新设备上拥有独立的权重。
• 如果模型加载器不支持多 GPU（没有重新加载工厂），该节点会原样传递模型并记录一条警告。
• 当 MultiGPU CFG Split 克隆已经占用所选设备时，该克隆会被移除，以免两个模型绑定到同一设备。
• 当选择非默认设备（CPU 或 GPU）时，该节点还会将模型的计算 dtype 调整为该设备支持的类型。
• 不建议将此节点放在已经使用过该模型的节点（例如 KSampler）之后，因为如果设备与原始设备匹配，前一个节点更改的任何状态都会被观察到。

输出

本文档由 AI 生成。如果您发现任何错误或有改进建议，欢迎贡献！ 在 GitHub 上编辑

Source fingerprint (SHA-256): d02a8bd9612861cf696f03969fe693088351de5a72ccbd4c1aed405b104eb71e