Initial support for RamTorch. Still a WIP
This commit is contained in:
1 parent
c6edd71a5b
commit
4e5707854f
8 files changed
+687
-120
No files matched your search
@@ -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