Victor Kuznetsov 675590e8b2 perf(invisible): read the fp16 weight variant to halve the cold-start weight load ...

InvisibleEngine loads SDXL/ControlNet in fp16 on CUDA/XPU but called from_pretrained
without variant="fp16", so it read the full fp32 weight files (~7 GB) and downcast in
memory. _load_from_pretrained now passes variant="fp16" when torch_dtype is float16,
reading the half-precision files (~3.5 GB) instead - roughly halving the cold-start
weight read + host->device transfer (a phase-timed Modal run measured weight load as
~half of the ~25s cold start). Falls back to the default weights when a checkpoint ships
no fp16 variant (a custom --model), so the worst case is the prior behavior. fp32
(cpu/mps) and bf16 (qwen) never request the variant.

Tests: TestFp16WeightVariant (variant requested on fp16, fallback on missing, never on
fp32).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

2026-06-23 14:42:47 -07:00

..

__init__.py

Add project files, tests, and documentation for GitHub release

2026-03-25 11:15:05 -07:00

conftest.py

Track the labeled SynthID corpus; complete metadata-source test coverage

2026-05-25 14:46:47 -07:00

test_cli.py

feat(invisible): skip the diffusion scrub when no invisible watermark is detectable (P0#5)

2026-06-22 11:37:01 -07:00

test_doubao_engine.py

feat(visible): add Jimeng remover, fix Doubao outline defect, reproducible mask build

2026-05-31 12:20:19 -07:00

test_fidelity_matching.py

fix(qwen): native-geometry img2img + pipeline-aware strength; record dropped auto/mixed/Z-Image leads

2026-06-20 21:52:56 -07:00

test_gemini_engine.py

fix(visible): inpaint mid-tone Gemini sparkle instead of a dark diamond

2026-06-18 17:19:41 -07:00

test_humanizer.py

feat(auto): adaptive detail-targeting polish + --adaptive-polish flag

2026-06-03 21:49:08 -07:00

test_identify.py

feat(invisible): skip the diffusion scrub when no invisible watermark is detectable (P0#5)

2026-06-22 11:37:01 -07:00

test_image_io.py

fix: address whole-project code review (visible all/batch, engine consolidation, I/O)

2026-06-09 13:21:13 -07:00

test_img2img_runner.py

feat: controlnet pipeline for text/face-structure preservation

2026-06-03 16:59:28 -07:00

test_invisible_engine.py

fix(invisible): preserve native output dimensions

2026-06-18 16:44:21 -07:00

test_invisible_watermark.py

feat(identify): detect open SD/SDXL/FLUX invisible watermark

2026-05-24 16:53:59 -07:00

test_jimeng_engine.py

feat(visible): add Jimeng remover, fix Doubao outline defect, reproducible mask build

2026-05-31 12:20:19 -07:00

test_metadata.py

feat(identify): detect + strip NovelAI / Reve / Aphrodite generator stamps

2026-06-22 16:23:33 -07:00

test_noai.py

feat(metadata): blank AI-generator tokens in AVIF/HEIF Exif meta-box items

2026-06-19 10:43:35 -07:00

test_optional_deps.py

chore: project review (dev tools in extras, dep upgrades, optional-deps guard, stale cleanup)

2026-06-09 17:03:17 -07:00

test_platform.py

perf(invisible): read the fp16 weight variant to halve the cold-start weight load

2026-06-23 14:42:47 -07:00

test_region_eraser.py

refactor: unify C2PA vendor registry + code-health fixes + uv publish

2026-06-03 21:01:07 -07:00

test_samsung_engine.py

feat(visible): Samsung Galaxy AI mark removal (bottom-left reverse-alpha, #37)

2026-06-05 10:27:44 -07:00

test_security_clamp.py

fix: harden metadata parsers and engines; sync docs (full-repo review)

2026-05-30 18:00:39 -07:00

test_synthid_corpus.py

feat(metadata): SynthID-source detection, C2PA parser consolidation, corpus + tests

2026-05-24 11:32:46 -07:00

test_synthid_pixel_probe.py

feat(probe): solid-fill SynthID carrier probe; corpus reconfirms no pixel detector

2026-05-24 16:35:39 -07:00

test_text_mark_memory.py

perf(text-mark): footprint-sized arrays in reverse-alpha CPU path

2026-06-19 10:01:07 -07:00

test_text_mark_oversubtraction.py

fix(visible): over-subtraction guard for Doubao/Jimeng/Samsung text marks

2026-06-20 15:34:39 -07:00

test_tiling.py

feat(invisible): region-targeted regeneration for AI-enhanced composites

2026-06-20 15:34:39 -07:00

test_trustmark_detector.py

Visible-watermark registry: reverse-alpha-only Doubao + Gemini, exact native recovery (#28)

2026-05-29 19:49:09 -07:00

test_upscaler.py

feat(auto): DBNet text detector, Real-ESRGAN upscaler, batch --auto

2026-06-04 16:04:33 -07:00

test_watermark_registry.py

feat(visible): Samsung Galaxy AI mark removal (bottom-left reverse-alpha, #37)

2026-06-05 10:27:44 -07:00