File size: 1,038 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
def is_integer(string):
try:
int(string)
return True
except ValueError:
return False
class HYAttnOverrideNode:
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"double_blocks": ("STRING", { "multiline": True }),
"single_blocks": ("STRING", { "multiline": True }),
}
}
RETURN_TYPES = ("ATTN_OVERRIDE",)
FUNCTION = "build"
CATEGORY = "hunyuanloom/attn"
def build(self, double_blocks, single_blocks):
double_block_map = set()
for block in double_blocks.split(','):
block = block.strip()
if is_integer(block):
double_block_map.add(int(block))
single_block_map = set()
for block in single_blocks.split(','):
block = block.strip()
if is_integer(block):
single_block_map.add(int(block))
return ({ "double": double_block_map, "single": single_block_map },)
|