ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 02:31:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added initial support for training i2v adapter WIP

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-09 08:06:29 -06:00
parent a8680c75eb
commit 615b0d0e94
7 changed files with 682 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
jobs/process/BaseSDTrainProcess.py
View File

@@ -321,9 +321,17 @@ class BaseSDTrainProcess(BaseTrainProcess):
if self.ema is not None:
self.ema.eval()
# let adapter know we are sampling
if self.adapter is not None and isinstance(self.adapter, CustomAdapter):
self.adapter.is_sampling = True
# send to be generated
self.sd.generate_images(gen_img_config_list, sampler=sample_config.sampler)
if self.adapter is not None and isinstance(self.adapter, CustomAdapter):
self.adapter.is_sampling = False
if self.ema is not None:
self.ema.train()
@@ -579,7 +587,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
direct_save = True
if self.adapter_config.type == 'redux':
direct_save = True
if self.adapter_config.type in ['control_lora', 'subpixel']:
if self.adapter_config.type in ['control_lora', 'subpixel', 'i2v']:
direct_save = True
save_ip_adapter_from_diffusers(
state_dict,