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

Added support for training flux redux adapters

Browse Source
This commit is contained in:
Jaret Burkett
2024-11-21 20:01:52 -07:00
parent 96d418bb95
commit 67c2e44edb
3 changed files with 88 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
jobs/process/BaseSDTrainProcess.py
View File

@@ -513,12 +513,17 @@ class BaseSDTrainProcess(BaseTrainProcess):
# move it back
self.adapter = self.adapter.to(orig_device, dtype=orig_dtype)
else:
direct_save = False
if self.adapter_config.train_only_image_encoder:
direct_save = True
if self.adapter_config.type == 'redux':
direct_save = True
save_ip_adapter_from_diffusers(
state_dict,
output_file=file_path,
meta=save_meta,
dtype=get_torch_dtype(self.save_config.dtype),
direct_save=self.adapter_config.train_only_image_encoder
direct_save=direct_save
)
else:
if self.save_config.save_format == "diffusers":