Junst
9a6b6ce77a
- 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)
10 lines
300 B
Python
10 lines
300 B
Python
# only import if running as a custom node
|
|
try:
|
|
import comfy.utils
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
from .nodes import NODE_CLASS_MAPPINGS
|
|
NODE_DISPLAY_NAME_MAPPINGS = {k:v.TITLE for k,v in NODE_CLASS_MAPPINGS.items()}
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|