ComfyUI Extension: ComfyUI-Torch-Compile

Authored by yondonfu

Created

Updated

3 stars

ComfyUI nodes for torch.compile.

Custom Nodes (0)

    README

    ComfyUI-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.


    Install

    Prererquisites

    The recommended installation method is to use the Comfy Registry.

    Comfy Registry

    These nodes can be installed via the Comfy Registry.

    comfy node registry-install comfyui-torch-compile
    

    ComfyUI-Manager

    These nodes can be installed via ComfyUI-Manager in the UI or via the CLI:

    comfy node install comfyui-torch-compile
    

    Manual

    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
    

    Nodes

    | 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. |