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

Change auto_memory to be layer_offloading and allow you to set the amount to unload

Browse Source
This commit is contained in:
Jaret Burkett
2025-10-10 13:12:32 -06:00
parent 2c2fbf16ea
commit 1bc6dee127
11 changed files with 279 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/BaseSDTrainProcess.py
View File

@@ -1759,7 +1759,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
)
# we cannot merge in if quantized
if self.model_config.quantize or self.model_config.auto_memory:
if self.model_config.quantize or self.model_config.layer_offloading:
# todo find a way around this
self.network.can_merge_in = False