Nodes/ComfyUI-AutoMultiGPU/Auto Checkpoint Loader (MultiGPU)
ComfyUI Node

Auto Checkpoint Loader (MultiGPU)

Pure PyTorch native checkpoint loader with automatic multi-GPU distribution.

By nexusfinancial-dev·Created 4 days ago·Updated 4 days ago· 1
Auto Checkpoint Loader (MultiGPU)
    • MODEL
    • CLIP
    • VAE
    ckpt_name
    distribution_modeAuto: UNet GPU0 / Encoders GPU1
    manual_unet_devicecpu
    manual_clip_devicecpu
    manual_vae_devicecpu
    CategoryAutoMultiGPU

    Inputs (5)

    NameTypeDefaultDescription
    ckpt_nameCOMBOThe checkpoint to load.
    distribution_modeCOMBOAuto: UNet GPU0 / Encoders GPU14 options: Auto: UNet GPU0 / Encoders GPU1, Auto: Balanced Free VRAM, Auto: Split Video (GPU0 Sampler / GPU1 VAE), Manual Override
    manual_unet_deviceoptCOMBOcpu1 options: cpu
    manual_clip_deviceoptCOMBOcpu1 options: cpu
    manual_vae_deviceoptCOMBOcpu1 options: cpu

    Outputs (3)

    NameTypeDescription
    MODELMODELModel (UNet/DiT) routed to Primary/Compute GPU.
    CLIPCLIPCLIP (Text Encoder) routed to Secondary GPU.
    VAEVAEVAE (Encode/Decode) routed to Secondary GPU.