Allow quantizing the te independently on flux. added lognorm_blend timestep schedule

2025-01-18 18:02:31 -07:00
parent 4723f23c0d
commit fadb2f3a76
3 changed files with 9 additions and 5 deletions
--- a/toolkit/samplers/custom_flowmatch_sampler.py
+++ b/toolkit/samplers/custom_flowmatch_sampler.py
@@ -110,7 +110,7 @@ class CustomFlowMatchEulerDiscreteScheduler(FlowMatchEulerDiscreteScheduler):
            return timesteps
        elif timestep_type == 'lognorm_blend':
            # disgtribute timestepd to the center/early and blend in linear
-            alpha = 0.8
+            alpha = 0.75

            lognormal = LogNormal(loc=0, scale=0.333)