VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove unnecessary device change

Browse Source
This commit is contained in:
Kohya S
2023-03-11 08:04:28 +09:00
parent 8b25929765
commit 0b38e663fd
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
networks/svd_merge_lora.py
View File

@@ -80,8 +80,6 @@ def merge_lora_models(models, ratios, new_rank, new_conv_rank, device, merge_dty
if device: # and isinstance(scale, torch.Tensor): if device: # and isinstance(scale, torch.Tensor):
scale = scale.to(device) scale = scale.to(device)
up_weight = up_weight.to(device)
down_weight = down_weight.to(device)
if not conv2d: # linear if not conv2d: # linear
weight = weight + ratio * (up_weight @ down_weight) * scale weight = weight + ratio * (up_weight @ down_weight) * scale