I added a second masking mode while keeping the original behavior available. With the original mode, the mask limits which reference tokens are used for feature transfer, but Klein can still see the full reference image for context.The new mode isolates the masked region more strictly. When connected unmasked reference tokens are blocked as attention sources, so the model only receives context from the selected area of that reference.This should be especially useful with multiple references. For example, one image can provide full identity context while another contributes only an outfit,, face, or other specific region. It should also help with outfit swaps, close-ups, and references containing distracting backgrounds or unrelated details. The full documentation explains how the node works and how the two masking modes differ. I recommend reading the masking section before testing it.I still recommend masking only what you need from the photos whether one or multiple as it give cleaner results :)Names of masking modes :Old behavior is focus_onlyNew behavior is zero_unmasked_tokens (recommended)The node's <a target="_blank" rel="ugc" href="https://github.com/capitan01R/ComfyUI-Flux2Klein-Enhancer#identity-feature-transfer-final">documentation</a>The mask behavior <a target="_blank" rel="ugc" href="https://github.com/capitan01R/ComfyUI-Flux2Klein-Enhancer#mask-behavior">documentation</a><hr />I updated <a target="_blank" rel="ugc" href="https://github.com/capitan01R/ComfyUI-Flux2Klein-Enhancer">Identity Feature Transfer</a> to remove the need for stacked/chained nodes.clearer <a target="_blank" rel="ugc" href="https://i.imgur.com/rYI6ZMi.png">screenshot</a> of the wf since reddit compresses the photosNow the workflow is simpler:<ul><li>Use Multi ReferenceLatent for multiple reference images.</li><li>Use Identity Feature Transfer Final for the identity pull.</li><li>If you use masks, connect each mask directly to the matching mask input on the node.<ul><li>subject_mask_1 = mask for reference 1</li><li>subject_mask_2 = mask for reference 2</li><li>etc.</li></ul></li></ul>The node handles the multi-reference setup internally, so you no longer need multiple stacked identity nodes for each reference.Presets are still available, similar to the previous version.For custom tuning, the two main knobs are:<ul><li>Temperature</li><li>Similarity</li></ul>Temperature is the main identity-strength control. Lower temperature gives a stronger, more direct 1:1 identity pull.Similarity works more like a refiner/filter. It controls how selective the match needs to be before the node pulls from the reference.So in practice:<ul><li>Lower temperature = stronger identity / more faithful match</li><li>Higher temperature = softer, looser identity influence</li><li>Lower similarity = allows more reference matches</li><li>Higher similarity = stricter matching, more selective pull</li></ul><a target="_blank" rel="ugc" href="https://github.com/capitan01R/ComfyUI-Flux2Klein-Enhancer/blob/main/example_workflow/Iden_feat_final_fixed.json">example workflow </a>(update to version 3.4.1 as there was a conflict with a node from a different repo causing the multireference latent node to be replaced if you had the other custom node installed and now that has been fixed)Also just a little side note, this Final version uses a bit diff technique in term of pulls so 1:1 is achievable but needs to be careful enough to get it.Previous posts for context: <a target="_blank" rel="ugc" href="https://www.reddit.com/r/StableDiffusion/comments/1tlmwzs/multi_referencelatent/">multi ref latent</a><a target="_blank" rel="ugc" href="https://www.reddit.com/r/StableDiffusion/comments/1t2ca6n/flux2_klein_identity_feature_transfer_v3_final/">Iden transfer v3</a>

Files with SHA256: 6a8b5d1ee288c79c0fb5fd1fe1f546f92c8b4fa723f4c707304e5680b58d8ed7