Add DPM++ 3M SDE DynETA

Partially implement PR #4 & #5 from blepping (oops hold on)
2024-03-04 12:32:26 -06:00
parent f3646e3790
commit 1f161b20f9
3 changed files with 154 additions and 71 deletions
@@ -14,5 +14,8 @@ NODE_CLASS_MAPPINGS = {
    "SamplerTTM": nodes.SamplerTTM,
    "SamplerLCMCustom": nodes.SamplerLCMCustom,
    "SamplerEulerAncestralDancing_Experimental": nodes.SamplerEULER_ANCESTRAL_DANCING,
+    "SamplerDPMPP_3M_SDE_DynETA": nodes.SamplerDPMPP_3M_SDE_DYN_ETA,
+    ### Schedulers
+    "SimpleExponentialScheduler": nodes.SimpleExponentialScheduler,
 }
 __all__ = ['NODE_CLASS_MAPPINGS']