xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-10 13:24:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_optimization_nodes.py

Browse Source
This commit is contained in:
kijai 2025-03-26 01:14:16 +02:00
parent b1ec996ba3
commit f77c0dc5d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -1204,7 +1204,7 @@ class CFGZeroStar:
dot_product = torch.sum(positive_flat * negative_flat, dim=1, keepdim=True) dot_product = torch.sum(positive_flat * negative_flat, dim=1, keepdim=True)
squared_norm = torch.sum(negative_flat ** 2, dim=1, keepdim=True) + 1e-8 squared_norm = torch.sum(negative_flat ** 2, dim=1, keepdim=True) + 1e-8
alpha = dot_product / squared_norm alpha = dot_product / squared_norm
alpha = alpha.view(batch_size, 1, 1, 1, 1) alpha = alpha.view(batch_size, *([1] * (len(cond.shape) - 1)))
matched_step_index = (sigmas == timestep[0] ).nonzero() matched_step_index = (sigmas == timestep[0] ).nonzero()
if len(matched_step_index) > 0: if len(matched_step_index) > 0: