DominikDoom/a1111-sd-webui-tagcomplete
1
0
Fork 0
mirror of https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git synced 2026-04-26 01:09:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into feature-sorting

Browse Source
This commit is contained in:
DominikDoom
2023-09-13 21:56:29 +02:00
parent 2846d79b7d 5067afeee9
commit f626b9453d
3 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/tag_autocomplete_helper.py
View File

@@ -89,7 +89,9 @@ def is_umi_format(data):
break
return not issue_found
def parse_umi_format(umi_tags, count, data):
count = 0
def parse_umi_format(umi_tags, data):
global count
for item in data:
umi_tags[count] = ','.join(data[item]['Tags'])
count += 1
@@ -119,7 +121,6 @@ def get_yaml_wildcards():
yaml_wildcards = {}
umi_tags = {} # { tag: count }
count = 0
for path in yaml_files:
try:
@@ -127,7 +128,7 @@ def get_yaml_wildcards():
data = yaml.safe_load(file)
if (data):
if (is_umi_format(data)):
parse_umi_format(umi_tags, count, data)
parse_umi_format(umi_tags, data)
else:
parse_dynamic_prompt_format(yaml_wildcards, data, path)
else: