Xinjie-Q commited on
Commit
8ea8ccd
·
verified ·
1 Parent(s): 89ea543

Silence content-policy banner; keep blank refusal image

Browse files

Remove print(verdict.banner()) in the t2i and edit paths so the classifier verdict is no longer written to stdout; blocked samples still return the plain white make_refusal_image placeholder.

Files changed (1) hide show
  1. mage_flow/pipeline.py +0 -2
mage_flow/pipeline.py CHANGED
@@ -286,7 +286,6 @@ def generate_images(model, prompts, neg_prompts=None, seeds=None, steps=30, cfg=
286
  verdict = model.txt_enc.screen_text(prompts[i])
287
  if verdict.violates:
288
  h_, w_ = _make_divisible_by_16(heights[i]), _make_divisible_by_16(widths[i])
289
- print(verdict.banner())
290
  results[i] = make_refusal_image(verdict, height=h_, width=w_)
291
  continue
292
  active.append(i)
@@ -477,7 +476,6 @@ def generate_edits(model, prompts, ref_images, neg_prompts=None, seeds=None, ste
477
  verdict = model.txt_enc.screen_edit(prompts[i], pils_per_sample[i])
478
  if verdict.violates:
479
  h_, w_ = res_hw[i]
480
- print(verdict.banner())
481
  results[i] = make_refusal_image(verdict, height=h_, width=w_)
482
  continue
483
  active.append(i)
 
286
  verdict = model.txt_enc.screen_text(prompts[i])
287
  if verdict.violates:
288
  h_, w_ = _make_divisible_by_16(heights[i]), _make_divisible_by_16(widths[i])
 
289
  results[i] = make_refusal_image(verdict, height=h_, width=w_)
290
  continue
291
  active.append(i)
 
476
  verdict = model.txt_enc.screen_edit(prompts[i], pils_per_sample[i])
477
  if verdict.violates:
478
  h_, w_ = res_hw[i]
 
479
  results[i] = make_refusal_image(verdict, height=h_, width=w_)
480
  continue
481
  active.append(i)