fix cosine rates

This commit is contained in:
Alex "mcmonkey" Goodwin
2023-02-09 22:46:49 -08:00
parent 0e324249e0
commit d838453c42
+2 -2
View File
@@ -126,11 +126,11 @@ class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiser):
elif mode == "Linear Down": elif mode == "Linear Down":
scale *= 1.0 - (self.step / max) scale *= 1.0 - (self.step / max)
elif mode == "Cosine Down": elif mode == "Cosine Down":
scale *= 1.0 - math.cos(self.step / max) scale *= 1.0 - math.cos((self.step / max) * 1.5707)
elif mode == "Linear Up": elif mode == "Linear Up":
scale *= self.step / max scale *= self.step / max
elif mode == "Cosine Up": elif mode == "Cosine Up":
scale *= math.cos(self.step / max) scale *= math.cos((self.step / max) * 1.5707)
scale += min scale += min
return scale return scale