ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add example config for qwen image edit

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-23 18:20:36 -06:00
parent 554dfb33bc
commit 5c27f89af5
4 changed files with 107 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/dataloader_mixins.py
View File

@@ -1865,7 +1865,9 @@ class TextEmbeddingCachingMixin:
self.sd.set_device_state_preset('cache_text_encoder')
did_move = True
if file_item.encode_control_in_text_embeddings and file_item.control_path is not None:
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")
# load the control image and feed it into the text encoder
ctrl_img = Image.open(file_item.control_path).convert("RGB")
# convert to 0 to 1 tensor