diff --git a/scripts/iib/db/update_image_data.py b/scripts/iib/db/update_image_data.py index 615b030..7f5c2a8 100644 --- a/scripts/iib/db/update_image_data.py +++ b/scripts/iib/db/update_image_data.py @@ -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: diff --git a/scripts/iib/parsers/fooocus.py b/scripts/iib/parsers/fooocus.py index 93120a6..af9e626 100644 --- a/scripts/iib/parsers/fooocus.py +++ b/scripts/iib/parsers/fooocus.py @@ -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(