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

Add new version of DFE. Kitchen sink

Browse Source
This commit is contained in:
Jaret Burkett
2025-01-31 11:42:27 -07:00
parent 34a1c6947a
commit 15a57bc89f
4 changed files with 203 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/BaseSDTrainProcess.py
View File

@@ -1636,7 +1636,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
if latest_save_path is not None:
state_dict = load_file(latest_save_path)
self.decorator.load_state_dict(state_dict)
self.load_training_state_from_metadata(path)
self.load_training_state_from_metadata(latest_save_path)
params.append({
'params': list(self.decorator.parameters()),