nodes/mcmonkeyprojects-sd-dynamic-thresholding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

correction to scale scheduler

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-01-31 20:29:06 -08:00
parent 1ea641b293
commit 6ba8969810
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+5 -4
View File
@@ -109,16 +109,17 @@ class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiser):
def interpretScale(self, scale, mode, min):
scale -= min
max = self.maxSteps - 1
if mode == "Constant":
pass
elif mode == "Linear Down":
scale *= 1.0 - (self.step / self.maxSteps)
scale *= 1.0 - (self.step / max)
elif mode == "Cosine Down":
scale *= 1.0 - math.cos(self.step / self.maxSteps)
scale *= 1.0 - math.cos(self.step / max)
elif mode == "Linear Up":
scale *= self.step / self.maxSteps
scale *= self.step / max
elif mode == "Cosine Up":
scale *= math.cos(self.step / self.maxSteps)
scale *= math.cos(self.step / max)
scale += min
return scale