1580158
/
sd-webui-infinite-image-browsing
mirror of https://github.com/zanllp/sd-webui-infinite-image-browsing.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Align metadata generated by various parsers

pull/563/head
zanllp 2024-03-27 21:03:59 +08:00
parent f2ec675775
commit 5c8ca65904
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/iib/db/update_image_data.py
View File

@ -96,6 +96,7 @@ def update_image_data(search_dirs: List[str], is_rebuild = False):
"Source Identifier", "Source Identifier",
"Postprocess upscale by", "Postprocess upscale by",
"Postprocess upscaler", "Postprocess upscaler",
"Size"
]: ]:
v = meta.get(k) v = meta.get(k)
if not v: if not v:

3
scripts/iib/parsers/fooocus.py
View File

@ -49,7 +49,8 @@ class FooocusParser:
v = v.replace(",", "，") v = v.replace(",", "，")
params["meta"][k] = v params["meta"][k] = v
metadata_list_str += f" {k}: {v}," metadata_list_str += f" {k}: {v},"
params["meta"]["Model"] = params["meta"]["Base Model"]
params["meta"]["Size"] = str(params["meta"]["Resolution"]).replace("(", "").replace(")", "").replace("，", " * ")
metadata_list_str = metadata_list_str.strip() metadata_list_str = metadata_list_str.strip()
info = f"""{params['Prompt']}\nNegative prompt: {params['Negative Prompt']}\n{metadata_list_str}""".strip() info = f"""{params['Prompt']}\nNegative prompt: {params['Negative Prompt']}\n{metadata_list_str}""".strip()
return ImageGenerationInfo( return ImageGenerationInfo(