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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-06-08 12:49:16 +03:00
parent 5430564e7f
commit cdd507de43
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/nodes.py
View File

@ -837,10 +837,11 @@ SVD:
def customsigmas(self, sigmas_string, interpolate_to_steps):
sigmas_list = sigmas_string.split(', ')
sigmas_float_list = [float(sigma) for sigma in sigmas_list]
sigmas_tensor = torch.tensor(sigmas_float_list)
if len(sigmas_tensor) != interpolate_to_steps:
sigmas_tensor = self.loglinear_interp(sigmas_tensor, interpolate_to_steps)
return (sigmas_tensor,)
sigmas_tensor = torch.FloatTensor(sigmas_float_list)
if len(sigmas_tensor) != interpolate_to_steps + 1:
sigmas_tensor = self.loglinear_interp(sigmas_tensor, interpolate_to_steps + 1)
sigmas_tensor[-1] = 0
return (sigmas_tensor.float(),)
def loglinear_interp(self, t_steps, num_steps):
"""