nodes/silveroxides-ComfyUI-ModelUtils
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix PowerUp namespace conflict

Browse Source
This commit is contained in:
silveroxides
2025-11-24 19:13:45 +01:00
parent 232dc76db2
commit 5ea63abfbe
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/merger_ops.py
+2 -2
View File
@@ -106,7 +106,7 @@ class Extract(Operation):
return (torch.lerp(a_f, b_f, self.alpha) * torch.lerp(d, 1 - d, self.beta)).to(a.dtype)
class Similarities(Extract):
def oper(self, a: torch.Tensor, b: torch.Tensor) -> torch.Tensor: return super().oper(None, a, b)
class PowerUp(Operation):
class PowerUpOp(Operation):
def __init__(self, key, alpha, seed, *sources):
super().__init__(key, *sources)
self.alpha, self.seed = alpha, seed
@@ -223,7 +223,7 @@ class PowerUp(CalcMode):
loader_args = {"handlers": kwargs['handlers'], "device": kwargs['device'], "dtype": kwargs['dtype']}
a = LoadTensor(key, kwargs['model_a'], **loader_args)
b = LoadTensor(key, kwargs['model_b'], **loader_args)
deltahat = PowerUp(key, kwargs['alpha'], kwargs['seed'], a, b)
deltahat = PowerUpOp(key, kwargs['alpha'], kwargs['seed'], a, b)
res = Multiply(key, kwargs['beta'], deltahat)
return Add(key, a, res)
class SVD(CalcMode):