ComfyUI Node: Realtime LoRA Trainer (Z-Image - Musubi Tuner)

Authored by shootthesound

Created

Updated

348 stars

Category

loaders

Inputs

inputcount INT
images_path STRING
musubi_path STRING
dit_model
    vae_model
      text_encoder
        caption STRING
        training_steps INT
        learning_rate FLOAT
        lora_rank INT
        vram_mode
        • Max (1256px)
        • Max (1256px) fp8
        • Max (1256px) fp8 offload
        • Medium (1024px)
        • Medium (1024px) fp8
        • Medium (1024px) fp8 offload
        • Low (768px)
        • Min (512px)
        keep_lora BOOLEAN
        output_name STRING
        custom_python_exe STRING
        image_1 IMAGE
        caption_1 STRING
        image_2 IMAGE
        caption_2 STRING
        image_3 IMAGE
        caption_3 STRING
        image_4 IMAGE
        caption_4 STRING

        Outputs

        STRING

        Extension: Realtime LoRA Trainer

        Train and Block Edit and Save LoRAs directly inside ComfyUI. Supports SDXL (via sd-scripts), FLUX, Z-Image Turbo, and Wan 2.2 (via Musubi Tuner and AI-Toolkit).

        Authored by shootthesound

        View Nodes

        Run ComfyUI workflows in the Cloud!

        No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

        Learn more