Update lora_nodes.py

This commit is contained in:
kijai 2025-09-08 10:43:50 +03:00
parent 6eb25e6b3d
commit 36f6fdd7d4

View File

@ -463,7 +463,7 @@ def resize_lora_model(lora_sd, new_rank, save_dtype, device, dynamic_method, dyn
total_keys = len([k for k in lora_sd if k.endswith(".weight")]) total_keys = len([k for k in lora_sd if k.endswith(".weight")])
pbar = comfy.utils.ProgressBar(total_keys) pbar = comfy.utils.ProgressBar(total_keys)
for key, value in tqdm(lora_sd.items(), leave=True, desc="Resizing LoRA weights", total=total_keys): for key, value in tqdm(lora_sd.items(), leave=True, desc="Resizing LoRA weights"):
key_parts = key.split(".") key_parts = key.split(".")
block_down_name = None block_down_name = None
for _format in LORA_DOWN_UP_FORMATS: for _format in LORA_DOWN_UP_FORMATS: