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