ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added specialized scaler training to ip adapters

Browse Source
This commit is contained in:
Jaret Burkett
2024-04-05 08:17:09 -06:00
parent 427847ac4c
commit 7284aab7c0
7 changed files with 182 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions_built_in/sd_trainer/SDTrainer.py
View File

@@ -1243,8 +1243,10 @@ class SDTrainer(BaseSDTrainProcess):
has_been_preprocessed=True,
quad_count=quad_count
)
# else:
# raise ValueError("Adapter images now must be loaded with dataloader or be a reg image")
else:
print("No Clip Image")
print([file_item.path for file_item in batch.file_items])
raise ValueError("Could not find clip image")
if not self.adapter_config.train_image_encoder:
# we are not training the image encoder, so we need to detach the embeds