Spaces:
Running on Zero
Running on Zero
Silence content-policy banner; keep blank refusal image
Browse filesRemove 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.
- 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)
|