Add node to allow getting BasicScheduler sigmas

This commit is contained in:
kijai
2025-07-29 23:06:01 +03:00
parent 1570287db1
commit 3f4e8433bd
+27 -2
View File
@@ -228,15 +228,40 @@ class CreateCFGScheduleFloatList:
return (cfg_list,)
class DummyComfyWanModelObject:
@classmethod
def INPUT_TYPES(s):
return {"required": {
}
}
RETURN_TYPES = ("MODEL", )
RETURN_NAMES = ("model",)
FUNCTION = "create"
CATEGORY = "WanVideoWrapper"
DESCRIPTION = "Helper node to create empty Wan model to use with BasicScheduler -node to get sigmas"
def create(self):
from comfy.model_sampling import ModelSamplingDiscreteFlow
class DummyModel:
def get_model_object(self, name):
if name == "model_sampling":
return ModelSamplingDiscreteFlow()
return None
return (DummyModel(),)
NODE_CLASS_MAPPINGS = {
"WanVideoImageResizeToClosest": WanVideoImageResizeToClosest,
"WanVideoVACEStartToEndFrame": WanVideoVACEStartToEndFrame,
"ExtractStartFramesForContinuations": ExtractStartFramesForContinuations,
"CreateCFGScheduleFloatList": CreateCFGScheduleFloatList
"CreateCFGScheduleFloatList": CreateCFGScheduleFloatList,
"DummyComfyWanModelObject": DummyComfyWanModelObject
}
NODE_DISPLAY_NAME_MAPPINGS = {
"WanVideoImageResizeToClosest": "WanVideo Image Resize To Closest",
"WanVideoVACEStartToEndFrame": "WanVideo VACE Start To End Frame",
"ExtractStartFramesForContinuations": "Extract Start Frames For Continuations",
"CreateCFGScheduleFloatList": "Create CFG Schedule Float List"
"CreateCFGScheduleFloatList": "Create CFG Schedule Float List",
"DummyComfyWanModelObject": "Dummy Comfy Wan Model Object"
}