ComfyUI-Torch-Compile is a set of ComfyUI nodes for running torch.compile.
These nodes are based off of the ones found in ComfyUI-KJNodes, but are standalone to separate torch.compile functionality from other nodes found in that pack and also fixes torch.compile for TAESD.
Prererquisites
- Install comfy-cli
The recommended installation method is to use the Comfy Registry.
These nodes can be installed via the Comfy Registry.
comfy node registry-install comfyui-torch-compile
These nodes can be installed via ComfyUI-Manager in the UI or via the CLI:
comfy node install comfyui-torch-compile
These nodes can also be installed manually by copying them into your custom_nodes folder and then installing dependencies:
cd custom_nodes
git clone https://github.com/yondonfu/ComfyUI-Torch-Compile
cd ComfyUI-Torch-Compile
pip install -r requirements.txt
| Node | Description |
|---|---|
| TorchCompileLoadVAE | Creates an optimized version of the VAE using torch.compile that will be JIT compiled during inference. |
| TorchCompileLoadControlNet | Creates an optimized version of the ControlNet using torch.compile that will be JIT compiled during inference. |