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

Merge branch 'comfyUI_improvements' of https://github.com/ermanitu/sd-webui-infinite-image-browsing into pr/ermanitu/772

pull/772/head
zanllp 2025-01-15 02:35:30 +08:00
parent d037420f91 82e0a0a4ff
commit 844debc55e
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/iib/tool.py
View File

@ -406,9 +406,8 @@ def get_comfyui_exif_data(img: Image):
return {} return {}
meta_key = '3' meta_key = '3'
data: Dict[str, any] = json.loads(prompt) data: Dict[str, any] = json.loads(prompt)
for i in range(3, 32): for i in data.keys():
try: try:
i = str(i)
if data[i]["class_type"].startswith("KSampler"): if data[i]["class_type"].startswith("KSampler"):
meta_key = i meta_key = i
break break
@ -416,13 +415,13 @@ def get_comfyui_exif_data(img: Image):
pass pass
meta = {} meta = {}
KSampler_entry = data[meta_key]["inputs"] KSampler_entry = data[meta_key]["inputs"]
print(KSampler_entry) #print(KSampler_entry) # for testing
# As a workaround to bypass parsing errors in the parser. # As a workaround to bypass parsing errors in the parser.
# https://github.com/jiw0220/stable-diffusion-image-metadata/blob/00b8d42d4d1a536862bba0b07c332bdebb2a0ce5/src/index.ts#L130 # https://github.com/jiw0220/stable-diffusion-image-metadata/blob/00b8d42d4d1a536862bba0b07c332bdebb2a0ce5/src/index.ts#L130
meta["Steps"] = KSampler_entry.get("steps", "Unknown") meta["Steps"] = KSampler_entry.get("steps", "Unknown")
meta["Sampler"] = KSampler_entry["sampler_name"] meta["Sampler"] = KSampler_entry["sampler_name"]
meta["Model"] = data[KSampler_entry["model"][0]]["inputs"].get("ckpt_name") # ermanitu meta["Model"] = data[KSampler_entry["model"][0]]["inputs"].get("ckpt_name")
meta["Source Identifier"] = "ComfyUI" meta["Source Identifier"] = "ComfyUI"
def get_text_from_clip(idx: str) : def get_text_from_clip(idx: str) :
inputs = data[idx]["inputs"] inputs = data[idx]["inputs"]
@ -431,7 +430,6 @@ def get_comfyui_exif_data(img: Image):
text = data[text[0]]["inputs"]["text"] text = data[text[0]]["inputs"]["text"]
return text.strip() return text.strip()
# added by ermanitu
in_node = data[str(KSampler_entry["positive"][0])] in_node = data[str(KSampler_entry["positive"][0])]
if in_node["class_type"] != "FluxGuidance": if in_node["class_type"] != "FluxGuidance":
pos_prompt = get_text_from_clip(KSampler_entry["positive"][0]) pos_prompt = get_text_from_clip(KSampler_entry["positive"][0])