External/SnapOtter
1
0
Fork 0
mirror of https://github.com/snapotter-hq/SnapOtter.git synced 2026-08-03 07:46:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(erase-object): optional high-quality diffusion inpainting bundle (#566)

Browse Source 
Adds an opt-in High Quality mode to the Object Eraser, backed by a new inpaint-hq feature bundle (Stable Diffusion 1.5 inpainting via diffusers). The default fast LaMa path is unchanged. Both arch archives are published to deepsafe/feature-bundles and the manifest carries their real sha256/sizes.

Verified end to end: a fresh container pulls the bundle from HuggingFace, checksum-verifies it, extracts torch/diffusers plus the fp16 model, and the HQ sidecar erases a large object with a plausible fill.

Refs #141
This commit is contained in:
SnapOtter
2026-07-19 20:47:35 +08:00
committed by GitHub
parent 84c18eb82c
commit 1bac663a2e
41 changed files with 954 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/e2e/erase-object.spec.ts
+38
View File
@@ -231,4 +231,42 @@ test.describe("Erase Object tool", () => {
// Both files now have masks -> batch submit button.
await expect(page.getByTestId("erase-object-submit")).toHaveText("Erase All (2)");
});
test("High Quality mode is gated on the inpaint-hq pack and blocks submit until installed", async ({
loggedInPage: page,
}) => {
// gotoEraser mocks only object-eraser-colorize as installed, so the optional
// inpaint-hq (diffusion) pack reads as missing.
await gotoEraser(page);
await uploadFile(page, fixturePath("image/valid/test-200x150.png"));
// The Fast/High-Quality toggle is present; Fast is the default.
await expect(page.getByTestId("eraser-quality-fast")).toHaveAttribute("aria-pressed", "true");
await expect(page.getByTestId("eraser-quality-hq")).toBeVisible();
// Paint a stroke so the ONLY thing gating submit is the quality mode.
const canvas = page.locator("canvas");
await canvas.waitFor({ state: "visible", timeout: 5_000 });
const box = await canvas.boundingBox();
if (!box) throw new Error("Canvas not found");
await page.mouse.move(box.x + box.width / 2, box.y + box.height / 2);
await page.mouse.down();
await page.mouse.move(box.x + box.width / 2 + 30, box.y + box.height / 2);
await page.mouse.up();
// Fast mode with a stroke: submit is enabled.
await expect(page.getByTestId("erase-object-submit")).toBeEnabled();
// Switch to High Quality: the pack is missing, so submit is blocked (never a
// silent downgrade to the fast path) and the install prompt appears.
await page.getByTestId("eraser-quality-hq").click();
await expect(page.getByTestId("eraser-quality-hq")).toHaveAttribute("aria-pressed", "true");
await expect(page.getByTestId("erase-object-submit")).toBeDisabled();
await expect(page.getByTestId("eraser-install-hq")).toBeVisible();
// Back to Fast: submit re-enables and the prompt is gone.
await page.getByTestId("eraser-quality-fast").click();
await expect(page.getByTestId("erase-object-submit")).toBeEnabled();
await expect(page.getByTestId("eraser-install-hq")).toHaveCount(0);
});
});