ComfyUI Node
CV VRAM Guard
Decide whether a heavy CV op should run on cuda or cpu based on VRAM usage.
CV VRAM Guard
- device_str
- warning
◄vram_used_mb0►
◄threshold_mb500►
◄device▾►
◄vram_total_mb0►
CategoryParametric/Image
Inputs (4)
| Name | Type | Default | Description |
|---|---|---|---|
| vram_used_mb | FLOAT | 00–1000000000 | Vram used mb. From 0 to 1e+09. Default 0.0. |
| threshold_mb | INT | 5000–1000000 | Threshold mb. From 0 to 1e+06. Default 500. |
| device | COMBO | Device. One of 3 choices. | |
| vram_total_mbopt | FLOAT | 00–1000000000 | Vram total mb. From 0 to 1e+09. Default 0.0. |
Outputs (2)
| Name | Type | Description |
|---|---|---|
| device_str | STRING | — |
| warning | STRING | — |