Nodes/Realtime LoRA Trainer/Selective LoRA Loader (Krea 2)
ComfyUI Node

Selective LoRA Loader (Krea 2)

The no-frills way to cut a Krea 2 LoRA's problem blocks out

By shootthesound·Created 9 months ago·Updated 21 days ago· 555
Selective LoRA Loader (Krea 2)
  • model
  • clip
  • model
  • clip
  • info
  • weights_output
lora_name
strength1.00
presetAll Blocks
block_0true
block_0_str1.00
block_1true
block_1_str1.00
block_2true
block_2_str1.00
block_3true
block_3_str1.00
block_4true
block_4_str1.00
block_5true
block_5_str1.00
block_6true
block_6_str1.00
block_7true
block_7_str1.00
block_8true
block_8_str1.00
block_9true
block_9_str1.00
block_10true
block_10_str1.00
block_11true
block_11_str1.00
block_12true
block_12_str1.00
block_13true
block_13_str1.00
block_14true
block_14_str1.00
block_15true
block_15_str1.00
block_16true
block_16_str1.00
block_17true
block_17_str1.00
block_18true
block_18_str1.00
block_19true
block_19_str1.00
block_20true
block_20_str1.00
block_21true
block_21_str1.00
block_22true
block_22_str1.00
block_23true
block_23_str1.00
block_24true
block_24_str1.00
block_25true
block_25_str1.00
block_26true
block_26_str1.00
block_27true
block_27_str1.00
other_weightstrue
other_weights_str1.00
lora_path_opt
analysis_json
block_weights_string

Krea 2 trained a LoRA that's perfect except for one recurring artifact, and you'd rather not go back for another 45-minute training run? This is the node you reach for. It's the plain selective loader from the ShootTheSound/comfyUI-Realtime-Lora pack ("Realtime LoRA Trainer"): no analysis built in, no strength scheduling, no saving - just a row of 28 checkboxes and sliders that decide which parts of your LoRA actually get applied.

The idea, in case you haven't hit this concept yet: a LoRA's effect is spread across the model's blocks, and not evenly. Some blocks learned the subject's face, some the style, some are mostly noise that only shows up as artifacts at specific strengths. Being able to flip a block off - or run it at 0.4 instead of 1.0 - is the difference between a LoRA that's usable and one that's bin-bound. Krea 2's 12B DiT has 28 main SingleStreamBlocks, which is exactly the 28 checkboxes you get here.

How it works

Boring in the good way. The node reads your LoRA file straight out of models/loras, works out which block each tensor belongs to, keeps only the blocks you've enabled (scaling each by its own _str slider), and hands the filtered result to ComfyUI's standard LoRA loader. Blocks 0–8 are the early ones, 9–18 mid, 19–27 late - the late blocks tend to carry the identity-heavy stuff, so start your experiments there. The Krea 2 modules that don't map to a SingleStreamBlock - first, last.linear, tmlp, txtmlp, tproj, txtfusion - all fall under one other_weights toggle.

The inputs that matter

  • model and clip - straight from your checkpoint and text encoder loaders, exactly like the stock LoraLoader.
  • preset - All Blocks by default, but the presets are where this earns its keep: Late Only (21-27), Mid-Late (14-27), Skip Early (7-27). These are one-click experiments that cost you nothing to try.
  • analysis_json - the one optional input worth knowing. Run the pack's "LoRA Loader + Analyzer" first and connect its JSON here; the checkboxes color-code by measured impact, so you're not guessing which blocks matter.
  • lora_path_opt - connects from an analyzer (or a trainer, after a fresh train) and overrides the dropdown.

Outputs are model, clip, plus two strings: info (a summary of what got loaded, filtered, and scaled) and weights_output (the block profile, if you want to reuse it as text elsewhere).

Installing it

cd ComfyUI/custom_nodes
git clone https://github.com/ShootTheSound/comfyUI-Realtime-Lora

Restart ComfyUI. You can also grab it through ComfyUI Manager by searching "Realtime LoRA Trainer". Important: unlike the training nodes in this same pack, the loaders and analyzers need nothing extra installed - no sd-scripts, no Musubi Tuner, no AI-Toolkit. The requirements file is literally empty beyond what ComfyUI already ships. If you just want to load and tweak Krea 2 LoRAs, this is a one-command install.

Where people get burned

  • Toggling everything off except other_weights produces garbage output. Disable in ranges, not all-or-nothing.
  • block_weights_string, if you connect one, silently overrides every checkbox you set. It's a power-user feature; ignore it until you understand it.
  • A selective loader can't fix a LoRA that's overcooked everywhere - block filtering is for removing specific, localized effects. If the whole thing's wrong, lower overall strength or retrain.

If you want the fancy version of this - the one that also shows you the impact scores, schedules strength over the generation, and can save the filtered result as a new LoRA - get the Krea 2 Analyzer + Selective Loader V2 from the same pack instead. But if you just want a checkbox to make the artifact go away, this is the one.

Categoryloaders/lora

Inputs (66)

NameTypeDefaultDescription
modelMODEL
clipCLIP
lora_nameCOMBOKrea 2 LoRA file to load
strengthFLOAT1.00-5–5Overall LoRA strength
presetCOMBOAll BlocksQuick preset selection. Choose 'Custom' to use individual toggles below.
block_0BOOLEANtrue
block_0_strFLOAT1.00-5–5
block_1BOOLEANtrue
block_1_strFLOAT1.00-5–5
block_2BOOLEANtrue
block_2_strFLOAT1.00-5–5
block_3BOOLEANtrue
block_3_strFLOAT1.00-5–5
block_4BOOLEANtrue
block_4_strFLOAT1.00-5–5
block_5BOOLEANtrue
block_5_strFLOAT1.00-5–5
block_6BOOLEANtrue
block_6_strFLOAT1.00-5–5
block_7BOOLEANtrue
block_7_strFLOAT1.00-5–5
block_8BOOLEANtrue
block_8_strFLOAT1.00-5–5
block_9BOOLEANtrue
block_9_strFLOAT1.00-5–5
block_10BOOLEANtrue
block_10_strFLOAT1.00-5–5
block_11BOOLEANtrue
block_11_strFLOAT1.00-5–5
block_12BOOLEANtrue
block_12_strFLOAT1.00-5–5
block_13BOOLEANtrue
block_13_strFLOAT1.00-5–5
block_14BOOLEANtrue
block_14_strFLOAT1.00-5–5
block_15BOOLEANtrue
block_15_strFLOAT1.00-5–5
block_16BOOLEANtrue
block_16_strFLOAT1.00-5–5
block_17BOOLEANtrue
block_17_strFLOAT1.00-5–5
block_18BOOLEANtrue
block_18_strFLOAT1.00-5–5
block_19BOOLEANtrue
block_19_strFLOAT1.00-5–5
block_20BOOLEANtrue
block_20_strFLOAT1.00-5–5
block_21BOOLEANtrue
block_21_strFLOAT1.00-5–5
block_22BOOLEANtrue
block_22_strFLOAT1.00-5–5
block_23BOOLEANtrue
block_23_strFLOAT1.00-5–5
block_24BOOLEANtrue
block_24_strFLOAT1.00-5–5
block_25BOOLEANtrue
block_25_strFLOAT1.00-5–5
block_26BOOLEANtrue
block_26_strFLOAT1.00-5–5
block_27BOOLEANtrue
block_27_strFLOAT1.00-5–5
other_weightsBOOLEANtrue
other_weights_strFLOAT1.00-5–5
lora_path_optoptSTRINGOptional: Connect from LoRA Analyzer to use its selected LoRA
analysis_jsonoptSTRINGOptional: Connect from LoRA Analyzer for impact-colored checkboxes
block_weights_stringoptSTRINGInput/Output block profile string. Positional text syncs with the UI. String input overrides UI values.

Outputs (4)

NameTypeDescription
modelMODEL
clipCLIP
infoSTRING
weights_outputSTRING