github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix issue with mask generator revision

Browse Source
This commit is contained in:
Jaret Burkett
2026-06-22 08:19:27 -06:00
parent d51463ca52
commit c133c55cf5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/control_generator.py
View File

@@ -156,7 +156,7 @@ class ControlGenerator:
self.control_bg_remover = AutoModelForImageSegmentation.from_pretrained( self.control_bg_remover = AutoModelForImageSegmentation.from_pretrained(
'ZhengPeng7/BiRefNet_HR', 'ZhengPeng7/BiRefNet_HR',
trust_remote_code=True, trust_remote_code=True,
revision="595e212b3eaa6a1beaad56cee49749b1e00b1596", revision="a7a562f6fd16021180f2f4348f4de003a2d3d1e1",
torch_dtype=torch.float16 torch_dtype=torch.float16
).to(device) ).to(device)
self.control_bg_remover.eval() self.control_bg_remover.eval()