diff --git a/src/remove_ai_watermarks/classify.py b/src/remove_ai_watermarks/classify.py index a041fa4..e3a88d3 100644 --- a/src/remove_ai_watermarks/classify.py +++ b/src/remove_ai_watermarks/classify.py @@ -21,7 +21,7 @@ log = logging.getLogger(__name__) CLASSIFY_EXTRA = "'remove-ai-watermarks[classify]'" WEIGHTS_ENV = "RAIW_CLASSIFY_WEIGHTS" WEIGHTS_REPO = "wiltodelta/raiw-photo-classify" -WEIGHTS_REVISION = "c0ac82b6f1ae9fc0b92c467562282e9422da6da6" +WEIGHTS_REVISION = "ffc46db5135ee3a83f51538f2c8f7483b9b8b40c" CLIP_FILE = "clip-l-ft.pt" PROBE_FILE = "probe-weights-clip-l-ft.npz" DETECTOR_FILE = "detector.pt" diff --git a/tests/test_classify.py b/tests/test_classify.py index ae95aed..fcaa6d4 100644 --- a/tests/test_classify.py +++ b/tests/test_classify.py @@ -49,7 +49,7 @@ def _scores( def test_hub_snapshot_is_the_freeze_revision() -> None: assert WEIGHTS_REPO == "wiltodelta/raiw-photo-classify" - assert WEIGHTS_REVISION == "c0ac82b6f1ae9fc0b92c467562282e9422da6da6" + assert WEIGHTS_REVISION == "ffc46db5135ee3a83f51538f2c8f7483b9b8b40c" def test_shipped_operating_point_matches_the_runtime_defaults() -> None: