Update nodes.py

2024-03-17 17:40:06 +01:00
parent 318f231d11
commit 4b73c13bfa
1 changed files with 1 additions and 1 deletions
@@ -43,7 +43,7 @@ def refine_token_weight(token_id, all_weights, subtract_difference, sculptor_mul
    del s[-1]
    del tmp_weights[-1]

-    if len(s) <= 1: return initial_weight.cpu(), 0
+    if len(s) < 1: return initial_weight.cpu(), 0

    concurrent_weights = torch.sum(torch.stack([t * s[i]**2 for i, t in enumerate(tmp_weights)]), dim=0)
    final_score = get_single_cosine_score(initial_weight,concurrent_weights) * sculptor_multiplier