mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
full codespell coverage
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -458,7 +458,7 @@ def apply_balanced_offload_to_module(module, op="apply", force:bool=False):
|
||||
module.balanced_offload_max_memory = max_memory
|
||||
module.offload_post = shared.sd_model_type in offload_post and module_name.startswith("text_encoder")
|
||||
if shared.opts.layerwise_quantization or getattr(module, 'quantization_method', None) == 'LayerWise':
|
||||
model_quant.apply_layerwise(module, quiet=True) # need to reapply since hooks were removed/readded
|
||||
model_quant.apply_layerwise(module, quiet=True) # need to reapply since hooks were removed/re-added
|
||||
devices.torch_gc(fast=True, force=True, reason='offload')
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user