alphard/ComfyUI-Lora-Auto-Trigger-Words
1
0
Fork 0
mirror of https://github.com/idrirap/ComfyUI-Lora-Auto-Trigger-Words.git synced 2026-03-21 21:22:12 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix : added boundaries for tag indexes and a check if tag list empty

Browse Source
This commit is contained in:
Dijkstra
2023-11-01 17:35:31 +01:00
parent 4f4cad063d
commit 13bbfbe616
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
utils.py
View File

@@ -171,12 +171,17 @@ def sort_tags_by_frequency(meta_tags):
return [] return []
def parse_selector(selector, tags_list): def parse_selector(selector, tags_list):
if len(tags_list) == 0:
return ""
range_index_list = selector.split(",") range_index_list = selector.split(",")
output = {} output = {}
for range_index in range_index_list: for range_index in range_index_list:
# single value # single value
if range_index.count(":") == 0: if range_index.count(":") == 0:
index = int(range_index) index = int(range_index)
# ignore out of bound indexes
if abs(index) > len(tags_list) - 1:
continue
output[index] = tags_list[index] output[index] = tags_list[index]
# actual range # actual range
@@ -196,6 +201,9 @@ def parse_selector(selector, tags_list):
start = len(tags_list) + start start = len(tags_list) + start
if end < 0: if end < 0:
end = len(tags_list) + end end = len(tags_list) + end
# clamp start and end values within list boundaries
start, end = min(start, len(tags_list)-1), min(end, len(tags_list)-1)
start, end = max(start, 0), max(end, 0)
# merge all # merge all
for i in range(start, end): for i in range(start, end):
output[i] = tags_list[i] output[i] = tags_list[i]