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

Added support to train Krea2 as an edit model

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-04 09:12:39 -06:00
parent f63221e577
commit b1e1a834d4
7 changed files with 519 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/dataloader_mixins.py
View File

@@ -1891,9 +1891,7 @@ class TextEmbeddingCachingMixin:
self.sd.set_device_state_preset('cache_text_encoder')
did_move = True
if file_item.encode_control_in_text_embeddings:
if file_item.control_path is None:
raise Exception(f"Could not find a control image for {file_item.path} which is needed for this model")
if file_item.encode_control_in_text_embeddings and file_item.control_path is not None:
ctrl_img_list = []
control_path_list = file_item.control_path
if not isinstance(file_item.control_path, list):