Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4952 from QualiaRain/fix/control-segmentanything-load

Browse Source 
fix(control): SegmentAnything 1.0 fails to load after processor rename
This commit is contained in:
Vladimir Mandic
2026-06-22 12:26:51 +02:00
committed by GitHub
parent 913f1ca9ee fdb27bb5af
commit 13be6d3f88
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/control/processors.py
+3 -3
View File
@@ -282,10 +282,10 @@ class Processor:
elif processor_id in ('DWPose (ONNX)', 'RTMW', 'RTMO'):
model_type = {'DWPose (ONNX)': 'DWPose', 'RTMW': 'RTMW-l', 'RTMO': 'RTMO-l'}[processor_id]
self.model = cls.from_pretrained(model_type, **self.load_config)
elif 'SegmentAnything' in processor_id:
if 'Base' == config['SegmentAnything']['model']:
elif processor_id == 'SegmentAnything 1.0':
if 'Base' == config[processor_id]['model']:
self.model = cls.from_pretrained(model_path = 'segments-arnaud/sam_vit_b', filename='sam_vit_b_01ec64.pth', model_type='vit_b', **self.load_config)
elif 'Large' == config['SegmentAnything']['model']:
elif 'Large' == config[processor_id]['model']:
self.model = cls.from_pretrained(model_path = 'segments-arnaud/sam_vit_l', filename='sam_vit_l_0b3195.pth', model_type='vit_l', **self.load_config)
else:
log.error(f'Control Processor load failed: id="{processor_id}" error=unknown model type')