diff --git a/nodes/lora_resize.py b/nodes/lora_resize.py index b15c3ea..a0c7646 100644 --- a/nodes/lora_resize.py +++ b/nodes/lora_resize.py @@ -1790,12 +1790,15 @@ def merge_loras_to_model( else: # No LoRA contribution, copy as-is output_sd[base_key] = cpu_base.to(save_dtype).contiguous() + del cpu_base stats["copied"] += 1 else: # Non-weight tensor (bias, norm, etc.), copy as-is output_sd[base_key] = cpu_base.to(save_dtype).contiguous() + del cpu_base stats["copied"] += 1 + pbar.update(1) if verbose: