Skip to content

Commit ab983ab

Browse files
committed
Update scheduler list
1 parent fbfc7d7 commit ab983ab

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

nodes_selectors.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
from typing import Any
2-
import comfy.sd
2+
import comfy
3+
4+
INSPIRE_SCHEDULERS = comfy.samplers.KSampler.SCHEDULERS + ['AYS SDXL', 'AYS SD1', 'AYS SVD', 'GITS[coeff=1.2]', 'LTXV[default]', 'OSS FLUX', 'OSS Wan', 'OSS Chroma']
5+
EFF_SCHEDULERS = comfy.samplers.KSampler.SCHEDULERS + ['AYS SD1', 'AYS SDXL', 'AYS SVD', 'GITS']
36

47
class SamplerSelector:
58
RETURN_TYPES = (comfy.samplers.KSampler.SAMPLERS, "STRING")
@@ -42,9 +45,9 @@ def get_names(self, scheduler: str) -> tuple[str, str]:
4245
return (scheduler, scheduler)
4346

4447
class SchedulerSelectorInspire:
45-
RETURN_TYPES = (comfy.samplers.KSampler.SCHEDULERS + ['AYS SDXL', 'AYS SD1', 'AYS SVD', 'GITS[coeff=1.2]', 'OSS FLUX', 'OSS Wan'], "STRING")
46-
RETURN_NAMES = ("scheduler", "scheduler_name")
47-
OUTPUT_TOOLTIPS = ("scheduler (SCHEDULERS + ['AYS SDXL', 'AYS SD1', 'AYS SVD', 'GITS[coeff=1.2]', 'OSS FLUX', 'OSS Wan'])", "scheduler name (STRING)")
48+
RETURN_TYPES = (INSPIRE_SCHEDULERS, "STRING")
49+
RETURN_NAMES = ("scheduler", "scheduler_name")
50+
OUTPUT_TOOLTIPS = ("scheduler (ComfyUI + Inspire Pack Schedulers)", "scheduler name (STRING)")
4851
FUNCTION = "get_names"
4952

5053
CATEGORY = 'ImageSaver/utils'
@@ -54,17 +57,17 @@ class SchedulerSelectorInspire:
5457
def INPUT_TYPES(cls) -> dict[str, Any]:
5558
return {
5659
"required": {
57-
"scheduler": (comfy.samplers.KSampler.SCHEDULERS + ['AYS SDXL', 'AYS SD1', 'AYS SVD', 'GITS[coeff=1.2]', 'OSS FLUX', 'OSS Wan'], {"tooltip": "scheduler (Comfy's standard + extras)"}),
60+
"scheduler": (INSPIRE_SCHEDULERS, {"tooltip": "scheduler (Comfy's standard + extras)"}),
5861
}
5962
}
6063

6164
def get_names(self, scheduler: str) -> tuple[str, str]:
6265
return (scheduler, scheduler)
6366

6467
class SchedulerSelectorEfficiency:
65-
RETURN_TYPES = (comfy.samplers.KSampler.SCHEDULERS + ['AYS SD1', 'AYS SDXL', 'AYS SVD', 'GITS'], "STRING")
66-
RETURN_NAMES = ("scheduler", "scheduler_name")
67-
OUTPUT_TOOLTIPS = ("scheduler (SCHEDULERS + ['AYS SD1', 'AYS SDXL', 'AYS SVD', 'GITS'])", "scheduler name (STRING)")
68+
RETURN_TYPES = (EFF_SCHEDULERS, "STRING")
69+
RETURN_NAMES = ("scheduler", "scheduler_name")
70+
OUTPUT_TOOLTIPS = ("scheduler (ComfyUI + Efficiency Pack Schedulers)", "scheduler name (STRING)")
6871
FUNCTION = "get_names"
6972

7073
CATEGORY = 'ImageSaver/utils'
@@ -74,7 +77,7 @@ class SchedulerSelectorEfficiency:
7477
def INPUT_TYPES(cls) -> dict[str, Any]:
7578
return {
7679
"required": {
77-
"scheduler": (comfy.samplers.KSampler.SCHEDULERS + ['AYS SD1', 'AYS SDXL', 'AYS SVD', 'GITS'], {"tooltip": "scheduler (Comfy's standard + Efficiency nodes)"}),
80+
"scheduler": (EFF_SCHEDULERS, {"tooltip": "scheduler (Comfy's standard + Efficiency nodes)"}),
7881
}
7982
}
8083

0 commit comments

Comments
 (0)