Change auto_memory to be layer_offloading and allow you to set the amount to unload
This commit is contained in:
1 parent
2c2fbf16ea
commit
1bc6dee127
11 files changed
+279
-45
No files matched your search
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user