From c133c55cf50826767f1686cdaaddf220e6637dcc Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Mon, 22 Jun 2026 08:19:27 -0600
Subject: [PATCH] Fix issue with mask generator revision

---
 toolkit/control_generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/control_generator.py b/toolkit/control_generator.py
index cf7ddaf..ea9dd5d 100644
--- a/toolkit/control_generator.py
+++ b/toolkit/control_generator.py
@@ -156,7 +156,7 @@ class ControlGenerator:
                 self.control_bg_remover = AutoModelForImageSegmentation.from_pretrained(
                     'ZhengPeng7/BiRefNet_HR',
                     trust_remote_code=True,
-                    revision="595e212b3eaa6a1beaad56cee49749b1e00b1596",
+                    revision="a7a562f6fd16021180f2f4348f4de003a2d3d1e1",
                     torch_dtype=torch.float16
                 ).to(device)
                 self.control_bg_remover.eval()