nodes/Extraltodeus-Vector_Sculptor_ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
Extraltodeus
2024-08-28 07:29:07 +02:00
committed by GitHub
parent 6e76843e49
commit 8859ac29e9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+2 -2
View File
@@ -59,13 +59,13 @@ def refine_token_weight(token_id, all_weights, sculptor_method, sculptor_multipl
if sculptor_method == "maximum_absolute":
concurrent_weights = torch.stack([ini_w/torch.norm(ini_w)]+[t/torch.norm(t) for i, t in enumerate(tmp_weights)])
initial_weight = maximum_absolute_values(concurrent_weights)
initial_weight *= pre_mag / torch.norm(initial_weight)
initial_weight = initial_weight * pre_mag / torch.norm(initial_weight)
return initial_weight.cpu(), len(s)
elif sculptor_method == "add_minimum_absolute":
concurrent_weights = torch.stack([ini_w/torch.norm(ini_w)]+[t/torch.norm(t) for i, t in enumerate(tmp_weights)])
initial_weight_min = maximum_absolute_values(concurrent_weights, sculptor_method == "minimum_absolute")
initial_weight = ini_w + initial_weight_min * sculptor_multiplier
initial_weight *= pre_mag / torch.norm(initial_weight)
initial_weight = initial_weight * pre_mag / torch.norm(initial_weight)
return initial_weight.cpu(), len(s)
concurrent_weights = torch.sum(torch.stack([t * s[i]**2 for i, t in enumerate(tmp_weights)]), dim=0)