ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added ui sopport for multi control samples and datasets. Added qwen image edit 5209 to the ui

Browse Source
This commit is contained in:
Jaret Burkett
2025-09-25 11:10:02 -06:00
parent 454be0958a
commit 1069dee0e4
10 changed files with 574 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
toolkit/config_modules.py
View File

@@ -831,6 +831,21 @@ class DatasetConfig:
if self.control_path == '':
self.control_path = None
# handle multi control inputs from the ui. It is just easier to handle it here for a cleaner ui experience
control_path_1 = kwargs.get('control_path_1', None)
control_path_2 = kwargs.get('control_path_2', None)
control_path_3 = kwargs.get('control_path_3', None)
if any([control_path_1, control_path_2, control_path_3]):
control_paths = []
if control_path_1:
control_paths.append(control_path_1)
if control_path_2:
control_paths.append(control_path_2)
if control_path_3:
control_paths.append(control_path_3)
self.control_path = control_paths
# color for transparent reigon of control images with transparency
self.control_transparent_color: List[int] = kwargs.get('control_transparent_color', [0, 0, 0])
# inpaint images should be webp/png images with alpha channel. The alpha 0 (invisible) section will