ComfyUI Node
PyTorchToDevice
A ComfyUI node in ETK/pytorch with 3 inputs and 2 outputs.
PyTorchToDevice
- model
- tensor
- TORCH_MODEL
- TORCH_TENSOR
◄devicecpu►
CategoryETK/pytorch
Inputs (3)
| Name | Type | Default | Description |
|---|---|---|---|
| device | COMBO | cpu | 2 options: cuda, cpu |
| modelopt | TORCH_MODEL | — | |
| tensoropt | TORCH_TENSOR | — |
Outputs (2)
| Name | Type | Description |
|---|---|---|
| TORCH_MODEL | TORCH_MODEL | — |
| TORCH_TENSOR | TORCH_TENSOR | — |