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

This commit is contained in:
Jaret Burkett committed 2025-10-10 13:12:32 -06:00
1 parent 2c2fbf16ea
commit 1bc6dee127
11 files changed
+279 -45

No files matched your search

+1 -1
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