Skip to content

Commit 1eab76d

Browse files
committed
(Webui script) Fix info text not showing image reward score
1 parent 9f58d44 commit 1eab76d

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

sdwebui/image_reward.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,15 @@ def run(self, p, filter_out_low_scores, lower_score_limit):
9191
for img in gens:
9292
score = shared.image_reward_model.score(p.prompt, img)
9393
img.info["score"] = score
94+
msg = f"ImageReward Score: {score:.4f}"
9495
if img.info.get("parameters") is None:
95-
img.info["parameters"] = f"ImageReward Score: {score:.4f}"
96+
img.info["parameters"] = msg
9697
else:
97-
img.info["parameters"] += f"\n ImageReward Score: {score:.4f}"
98+
img.info["parameters"] += "\n " + msg
99+
if getattr(proc, 'info', None) is None:
100+
setattr(proc, 'info', msg)
101+
else:
102+
proc.info += "\n " + msg
98103

99104
# filter out images with scores lower than the lower limit
100105
if filter_out_low_scores:

0 commit comments

Comments
 (0)