Fix issue with mask generator revision

2026-06-22 08:19:27 -06:00
parent d51463ca52
commit c133c55cf5
1 changed files with 1 additions and 1 deletions
--- 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()