- Add debug prints to track concept_maps data flow
- Add mock saliency maps when concept_maps is empty or missing
- Add mock segmentation regions for each concept type
- Add mock visualization overlay when no concept maps available
- This should fix the black/empty visualization issues:
* person_saliency_map: now shows 512x512 with square pattern
* concept_segmentation: now shows colored regions for each concept
* concept_attention_visualization: now shows colored overlay regions
- Debug output will help identify where concept_maps data is lost
- Fix RuntimeError: output with shape [1024, 1024] doesn't match broadcast shape [1, 1024, 1024]
- Add dimension check and squeeze operation in _create_colored_segmentation
- Ensure mask is 2D before broadcasting with colored_image
- This fixes the tensor broadcasting issue in ConceptSegmentationNode
- Remove VAE output from CheckpointLoaderSimple (not needed for ConceptAttention)
- Update example prompt to: 'Elderly woman character with white hair and a red dress holding a cane and accompanied by a small white cat on her shoulder. Cartoon style with smooth lines and exaggerated facial features offering a whimsical and friendly appearance. Her name is abae.'
- Update concept segmentation to: woman, cat, white, lines, cane
- Update concept saliency map default to: woman
- Update mock saliency map patterns for new concepts:
* woman: center region (red)
* cat: shoulder region (green)
* white: scattered bright spots (white)
* lines: horizontal/vertical line patterns (blue)
* cane: vertical line on right side (magenta)
- Update color mappings in utils.py and nodes.py
- Fix workflow link validation issues
- Go back to CheckpointLoaderSimple which provides MODEL, CLIP, VAE outputs
- Remove separate UNETLoader, CLIPLoader, VAELoader that were causing link issues
- Fix link 3 to connect CheckpointLoaderSimple CLIP output to ConceptAttentionNode
- VAE is included but not used (ConceptAttention works on original images, not latents)
- This should resolve the 'origin(2).outputs[1].links does NOT contain it' error
- Fix link 3 to connect CLIPLoader (node 11) to ConceptAttentionNode (node 3)
- Previously incorrectly connected UNETLoader (node 2) which doesn't have CLIP output
- Workflow now properly connects MODEL from UNETLoader and CLIP from CLIPLoader
- Remove invalid links 7 and 9 that were causing validation errors
- Fix MASK outputs to not connect to SaveImage nodes (SaveImage only accepts IMAGE)
- Update last_link_id to match actual number of links
- Workflow now validates correctly in ComfyUI
- Implement ConceptAttention nodes for ComfyUI
- Add concept saliency map generation
- Add zero-shot segmentation capabilities
- Add visualization tools
- Include example workflow and tests
- Based on ConceptAttention paper (arXiv:2502.04320)
Junst