xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 03:07:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes_train.py

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-06-11 00:18:23 +08:00 committed by GitHub
parent b8757c5822
commit 4dcd698e48
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_train.py
View File

@ -429,13 +429,13 @@ class TrainLoraNode:
if existing_adapter is not None: if existing_adapter is not None:
train_adapter = existing_adapter.to_train().to(lora_dtype) train_adapter = existing_adapter.to_train().to(lora_dtype)
for name, parameter in train_adapter.named_parameters():
lora_sd[f"{n}.{name}"] = parameter
else: else:
# Use LoRA with alpha=1.0 by default # Use LoRA with alpha=1.0 by default
train_adapter = adapter_cls.create_train( train_adapter = adapter_cls.create_train(
m.weight, rank=rank, alpha=1.0 m.weight, rank=rank, alpha=1.0
).to(lora_dtype) ).to(lora_dtype)
for name, parameter in train_adapter.named_parameters():
lora_sd[f"{n}.{name}"] = parameter
mp.add_weight_wrapper(key, train_adapter) mp.add_weight_wrapper(key, train_adapter)
all_weight_adapters.append(train_adapter) all_weight_adapters.append(train_adapter)