def is_integer(string):
    try:
        int(string)
        return True
    except ValueError:
        return False
    

class LTXAttnOverrideNode:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "layers": ("STRING", { "multiline": True }),
            }
        }

    RETURN_TYPES = ("ATTN_OVERRIDE",)
    FUNCTION = "build"

    CATEGORY = "ltxtricks/attn"

    def build(self, layers):
        layers_map = set([])
        for block in layers.split(','):
            block = block.strip()
            if is_integer(block):
                layers_map.add(block)

        return ({ "layers": layers_map },)