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

Update sigmas_merge.py

Browse Source
This commit is contained in:
Extraltodeus
2023-12-07 19:10:54 +01:00
committed by GitHub
parent bef779c9a0
commit 39abf2dfb9
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sigmas_mix/sigmas_merge.py
+2 -3
View File
@@ -134,7 +134,7 @@ class the_golden_scheduler:
sigmax = s.sigma(s.timestep(s.sigma_max))
phi = (1 + 5 ** 0.5) / 2
sigmas = torch.tensor([(1-x/(steps-1))**phi*sigmax+(x/(steps-1))**phi*sigmin for x in range(steps)]+[0]).cuda()
sigmas = torch.tensor([(1-x/(steps-1))**phi*sigmax+(x/(steps-1))**phi*sigmin for x in range(steps)]+[0])
return (sigmas,)
class manual_scheduler:
@@ -171,7 +171,7 @@ class manual_scheduler:
print("could not evaluate {custom_sigmas_manual_schedule}")
f = 0
sigmas.append(f)
sigmas = torch.tensor(sigmas+[0]).cuda()
sigmas = torch.tensor(sigmas+[0])
return (sigmas,)
def remap_range_no_clamp(value, minIn, MaxIn, minOut, maxOut):
@@ -226,7 +226,6 @@ class sigmas_gradual_merge:
for idx,s in enumerate(result_sigmas):
current_factor = remap_range_no_clamp(idx,0,len(result_sigmas)-1,proportion_1,1-proportion_1)
result_sigmas[idx] = sigmas_1[idx]*current_factor+sigmas_2[idx]*(1-current_factor)
result_sigmas = result_sigmas.cuda()
return (result_sigmas,)
NODE_CLASS_MAPPINGS = {