xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 04:44:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-02-25 01:04:05 +02:00
parent a7382e40c5
commit 2441d098b8
1 changed files with 35 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
nodes.py
View File

@ -3659,7 +3659,37 @@ class ImageNormalize_Neg1_To_1:
images = images * 2.0 - 1.0
return (images,)
class X0_passlatent(comfy.model_sampling.EPS):
def calculate_denoised(self, sigma, model_output, model_input):
return model_output
def calculate_input(self, sigma, noise):
return noise
class SingleStepSampling:
@classmethod
def INPUT_TYPES(s):
return {"required": { "model": ("MODEL",),
}}
RETURN_TYPES = ("MODEL",)
FUNCTION = "patch"
CATEGORY = "KJNodes"
def patch(self, model):
m = model.clone()
sampling_base = comfy.model_sampling.ModelSamplingDiscrete
sampling_type = X0_passlatent
class ModelSamplingAdvanced(sampling_base, sampling_type):
pass
model_sampling = ModelSamplingAdvanced(model.model.model_config)
m.add_object_patch("model_sampling", model_sampling)
return (m, )
NODE_CLASS_MAPPINGS = {
"INTConstant": INTConstant,
"FloatConstant": FloatConstant,
@ -3727,7 +3757,8 @@ NODE_CLASS_MAPPINGS = {
"ImageUpscaleWithModelBatched": ImageUpscaleWithModelBatched,
"ScaleBatchPromptSchedule": ScaleBatchPromptSchedule,
"EffnetEncode": EffnetEncode,
"ImageNormalize_Neg1_To_1": ImageNormalize_Neg1_To_1
"ImageNormalize_Neg1_To_1": ImageNormalize_Neg1_To_1,
"SingleStepSampling": SingleStepSampling,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant",
@ -3795,5 +3826,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"ImageUpscaleWithModelBatched": "ImageUpscaleWithModelBatched",
"ScaleBatchPromptSchedule": "ScaleBatchPromptSchedule",
"EffnetEncode": "EffnetEncode",
"ImageNormalize_Neg1_To_1": "ImageNormalize_Neg1_To_1"
"ImageNormalize_Neg1_To_1": "ImageNormalize_Neg1_To_1",
"SingleStepSampling": "SingleStepSampling",
}