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

View File

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

View File

@ -49,7 +49,8 @@ class FooocusParser:
v = v.replace(",", "")
params["meta"][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()
info = f"""{params['Prompt']}\nNegative prompt: {params['Negative Prompt']}\n{metadata_list_str}""".strip()
return ImageGenerationInfo(