Ethanfel/ComfyUI-Prompt-Calibrator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Trim dead inputs: drop fp16 precision and prompt_used

Browse Source 
fp16 offers nothing over bf16 for these models (removed from the quant dropdown;
loader still tolerant if passed). prompt_used was metadata-only — removed from the
node inputs, report payload/markdown, the bridge, and the example workflows.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-06-27 10:03:06 +02:00
parent 271aa8ae42
commit d389d6daff
7 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/qwen_judge.py
+4 -7
View File
@@ -619,7 +619,7 @@ def _report_base_dir(report_dir: str) -> str:
return os.path.join(os.path.dirname(os.path.dirname(__file__)), "output", "calibrator")
def _write_report(report_dir, run_tag, overall, merged, diff_analysis, raw_all, prompt_used,
def _write_report(report_dir, run_tag, overall, merged, diff_analysis, raw_all,
mismatch_count=0):
"""Persist the analysis so the external CLI agent can read it after a queue.
@@ -638,7 +638,6 @@ def _write_report(report_dir, run_tag, overall, merged, diff_analysis, raw_all,
"mismatch_count": mismatch_count,
"axes": (merged or {}).get("axes", {}),
"diff_analysis": diff_analysis,
"prompt_used": prompt_used,
"raw": raw_all,
}
tag = re.sub(r"[^A-Za-z0-9._-]", "_", run_tag) if run_tag else "latest"
@@ -653,7 +652,6 @@ def _write_report(report_dir, run_tag, overall, merged, diff_analysis, raw_all,
try:
md = (f"# Calibration analysis ({tag})\n\n"
f"**overall_score:** {payload['overall_score']}\n\n"
f"**prompt_used:**\n\n{prompt_used or '(not provided)'}\n\n"
f"## per-axis\n\n{diff_analysis}\n")
with open(os.path.join(base, f"calib_{tag}.md"), "w", encoding="utf-8") as f:
f.write(md)
@@ -746,7 +744,7 @@ class QwenVLImageJudge:
"model_select": ([MANUAL_CHOICE] + list(MODEL_PRESETS.keys()),
{"default": list(MODEL_PRESETS.keys())[0]}),
"model_path": ("STRING", {"default": ""}), # manual override (local dir / HF repo / alias)
"precision": (["bf16", "fp16", "fp8", "nf4"], {"default": "bf16"}),
"precision": (["bf16", "fp8", "nf4"], {"default": "bf16"}),
"axes": ("STRING", {"default": "", "multiline": True}),
"max_new_tokens": ("INT", {"default": 1024, "min": 64, "max": 4096}),
"temperature": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.5, "step": 0.05}),
@@ -759,7 +757,6 @@ class QwenVLImageJudge:
# The agent reads the analysis from these files after each queue.
"report_dir": ("STRING", {"default": ""}),
"run_tag": ("STRING", {"default": ""}),
"prompt_used": ("STRING", {"default": "", "multiline": True}),
# compare: canonical reference text (from describe). When set, compare
# anchors on it instead of re-reading the reference image each time.
"reference_description": ("STRING", {"default": "", "multiline": True}),
@@ -773,7 +770,7 @@ class QwenVLImageJudge:
max_new_tokens, temperature, swap_eval, profile="general",
model_select=MANUAL_CHOICE, generated_image=None,
keep_loaded=True, auto_download=True,
report_dir="", run_tag="", prompt_used="", reference_description="",
report_dir="", run_tag="", reference_description="",
system_prompt="", user_prompt="Describe this image."):
# `axes` overrides the profile when provided; otherwise use the profile's axis set.
axis_list = [a.strip() for a in re.split(r"[,\n]", axes) if a.strip()]
@@ -866,7 +863,7 @@ class QwenVLImageJudge:
diff_analysis = header + "\n" + "\n".join(diff_lines) if diff_lines else "(no parseable judgement)"
report_path = _write_report(
report_dir, run_tag, overall, merged, diff_analysis, raw_all, prompt_used, mismatch_count)
report_dir, run_tag, overall, merged, diff_analysis, raw_all, mismatch_count)
return (round(overall, 4), axis_scores, diff_analysis, raw_all, report_path)