diff --git a/tests/test_head_auto_apply.py b/tests/test_head_auto_apply.py index 91314ce..0d345e0 100644 --- a/tests/test_head_auto_apply.py +++ b/tests/test_head_auto_apply.py @@ -35,7 +35,7 @@ def _img(db, sha: str, emb) -> ImageRecord: return img -def _head(db, tag_id: int, slot: int, *, threshold=0.5, n_pos=30): +def _head(db, tag_id: int, slot: int, *, threshold=0.5, n_pos=60): s = db.execute(select(MLSettings).where(MLSettings.id == 1)).scalar_one() w = [0.0] * 1152 w[slot] = 1.0