ComfyUI nodes for torch.compile.
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
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. |