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

More work on mean flow loss. Moved it to an adapter. Still not functioning properly though.

Browse Source
This commit is contained in:
Jaret Burkett
2025-06-16 07:17:35 -06:00
parent c0314ba325
commit 1c2b7298dd
6 changed files with 323 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
jobs/process/BaseSDTrainProcess.py
View File

@@ -575,10 +575,8 @@ class BaseSDTrainProcess(BaseTrainProcess):
direct_save = False
if self.adapter_config.train_only_image_encoder:
direct_save = True
if self.adapter_config.type == 'redux':
direct_save = True
if self.adapter_config.type in ['control_lora', 'subpixel', 'i2v']:
direct_save = True
elif isinstance(self.adapter, CustomAdapter):
direct_save = self.adapter.do_direct_save
save_ip_adapter_from_diffusers(
state_dict,
output_file=file_path,