mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-29 18:51:37 +00:00
Initial support for RamTorch. Still a WIP
This commit is contained in:
@@ -1759,7 +1759,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
|
||||
)
|
||||
|
||||
# we cannot merge in if quantized
|
||||
if self.model_config.quantize:
|
||||
if self.model_config.quantize or self.model_config.auto_memory:
|
||||
# todo find a way around this
|
||||
self.network.can_merge_in = False
|
||||
|
||||
|
||||
Reference in New Issue
Block a user