Add sapiens2 matting as a mask generator. Begin transition to model paths and model folders.

2026-05-20 08:56:16 -06:00
parent 15d4fb89ff
commit aecd554128
8 changed files with 209 additions and 34 deletions
--- a/toolkit/dataloader_mixins.py
+++ b/toolkit/dataloader_mixins.py
@@ -2297,7 +2297,7 @@ class ControlCachingMixin:
        if control_type == 'inpaint':
            file_item.inpaint_path = control_path
            file_item.has_inpaint_image = True
-        elif control_type == 'mask':
+        elif control_type == 'mask' or control_type == 'sapiens2_mask':
            file_item.mask_path = control_path
            file_item.has_mask_image = True
        else: