From 1ee4ecfe88f4aa29da3dbe81cd6d46595159cabd Mon Sep 17 00:00:00 2001 From: Junst Date: Wed, 24 Sep 2025 17:41:20 +0900 Subject: [PATCH] Fix concept_list string to list conversion - Convert concept_list string to proper list format - Handle both string and list inputs for concept_list - Fix character-by-character splitting issue - Ensure proper concept processing for attention maps - Improve concept extraction from prompt --- nodes.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/nodes.py b/nodes.py index 562859b..92b2b5f 100644 --- a/nodes.py +++ b/nodes.py @@ -42,6 +42,14 @@ class ConceptAttentionNode: Generate concept attention maps using simplified approach. """ try: + # Convert concept_list string to list + if isinstance(concept_list, str): + concept_list = [concept.strip() for concept in concept_list.split(',')] + elif isinstance(concept_list, list): + concept_list = [str(concept).strip() for concept in concept_list] + else: + concept_list = [] + logger.info(f"DEBUG: Extracted concepts from prompt: {concept_list}") # Initialize processor if not exists