Commit 69a824e9 authored by comfyanonymous's avatar comfyanonymous
Browse files

Move _for_testing/custom_sampling nodes to sampling/custom_sampling.

parent a0b1d4f2
...@@ -15,7 +15,7 @@ class BasicScheduler: ...@@ -15,7 +15,7 @@ class BasicScheduler:
} }
} }
RETURN_TYPES = ("SIGMAS",) RETURN_TYPES = ("SIGMAS",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -35,7 +35,7 @@ class KarrasScheduler: ...@@ -35,7 +35,7 @@ class KarrasScheduler:
} }
} }
RETURN_TYPES = ("SIGMAS",) RETURN_TYPES = ("SIGMAS",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -53,7 +53,7 @@ class ExponentialScheduler: ...@@ -53,7 +53,7 @@ class ExponentialScheduler:
} }
} }
RETURN_TYPES = ("SIGMAS",) RETURN_TYPES = ("SIGMAS",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -72,7 +72,7 @@ class PolyexponentialScheduler: ...@@ -72,7 +72,7 @@ class PolyexponentialScheduler:
} }
} }
RETURN_TYPES = ("SIGMAS",) RETURN_TYPES = ("SIGMAS",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -91,7 +91,7 @@ class VPScheduler: ...@@ -91,7 +91,7 @@ class VPScheduler:
} }
} }
RETURN_TYPES = ("SIGMAS",) RETURN_TYPES = ("SIGMAS",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -108,7 +108,7 @@ class SplitSigmas: ...@@ -108,7 +108,7 @@ class SplitSigmas:
} }
} }
RETURN_TYPES = ("SIGMAS","SIGMAS") RETURN_TYPES = ("SIGMAS","SIGMAS")
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sigmas" FUNCTION = "get_sigmas"
...@@ -125,7 +125,7 @@ class KSamplerSelect: ...@@ -125,7 +125,7 @@ class KSamplerSelect:
} }
} }
RETURN_TYPES = ("SAMPLER",) RETURN_TYPES = ("SAMPLER",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sampler" FUNCTION = "get_sampler"
...@@ -144,7 +144,7 @@ class SamplerDPMPP_2M_SDE: ...@@ -144,7 +144,7 @@ class SamplerDPMPP_2M_SDE:
} }
} }
RETURN_TYPES = ("SAMPLER",) RETURN_TYPES = ("SAMPLER",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sampler" FUNCTION = "get_sampler"
...@@ -168,7 +168,7 @@ class SamplerDPMPP_SDE: ...@@ -168,7 +168,7 @@ class SamplerDPMPP_SDE:
} }
} }
RETURN_TYPES = ("SAMPLER",) RETURN_TYPES = ("SAMPLER",)
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
FUNCTION = "get_sampler" FUNCTION = "get_sampler"
...@@ -201,7 +201,7 @@ class SamplerCustom: ...@@ -201,7 +201,7 @@ class SamplerCustom:
FUNCTION = "sample" FUNCTION = "sample"
CATEGORY = "_for_testing/custom_sampling" CATEGORY = "sampling/custom_sampling"
def sample(self, model, add_noise, noise_seed, cfg, positive, negative, sampler, sigmas, latent_image): def sample(self, model, add_noise, noise_seed, cfg, positive, negative, sampler, sigmas, latent_image):
latent = latent_image latent = latent_image
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment