ComfyUI Extension: ComfyUI-FluxRegionAttention

Authored by attashe

Created

Updated

102 stars

Implement Region Attention for Flux model. Add node RegionAttention that takes a regions - mask + condition, mask could be set from comfyui masks or bbox in FluxRegionBBOX node. This code is not optimized and has a memory leak. If you caught a OOM just try run a query againg - works on my RTX3080. For generation it uses a usual prompt that have influence to all picture and a regions that have their own prompts. Base prompt good for setup background and style of image. This is train-free technique and results not always stable - sometimes need to try several seeds or change prompt.

Custom Nodes (0)

    README

    ComfyUI-FluxRegionAttention

    Implement Region Attention for Flux model. Add node RegionAttention that takes a regions - mask + condition, mask could be set from comfyui masks or bbox in FluxRegionBBOX node. This code is not optimized and has a memory leak. If you caught a OOM just try run a query againg - works on my RTX3080.

    For generation it uses a usual prompt that have influence to all picture and a regions that have their own prompts. Base prompt good for setup background and style of image.

    This is train-free technique and results not always stable - sometimes need to try several seeds or change prompt.

    Examples

    Workflows in the metadata

    clown in the forest black swan colored brick wall

    Nodes screenshot

    1. Region attention influence only to t5_xxl embeddings, for clip_l embeddings we can use concatinated prompt (stronger regional conditioning) or only common prompg (weaker conditioning).

    concat prompt to clip_l background prompt to clip_l

    1. Pipeline

    colored wall black swan

    Aknowledgements

    This repository is base on next repositories:

    @Misc{omost, author = {Omost Team}, title = {Omost GitHub Page}, year = {2024}, }

    Gligen-GUI

    black-forest-labs

    lucidrains attention implementation