1
1
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' ]
3
6
4
7
class SamplerSelector :
5
8
RETURN_TYPES = (comfy .samplers .KSampler .SAMPLERS , "STRING" )
@@ -42,9 +45,9 @@ def get_names(self, scheduler: str) -> tuple[str, str]:
42
45
return (scheduler , scheduler )
43
46
44
47
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)" )
48
51
FUNCTION = "get_names"
49
52
50
53
CATEGORY = 'ImageSaver/utils'
@@ -54,17 +57,17 @@ class SchedulerSelectorInspire:
54
57
def INPUT_TYPES (cls ) -> dict [str , Any ]:
55
58
return {
56
59
"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)" }),
58
61
}
59
62
}
60
63
61
64
def get_names (self , scheduler : str ) -> tuple [str , str ]:
62
65
return (scheduler , scheduler )
63
66
64
67
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)" )
68
71
FUNCTION = "get_names"
69
72
70
73
CATEGORY = 'ImageSaver/utils'
@@ -74,7 +77,7 @@ class SchedulerSelectorEfficiency:
74
77
def INPUT_TYPES (cls ) -> dict [str , Any ]:
75
78
return {
76
79
"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)" }),
78
81
}
79
82
}
80
83
0 commit comments