jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Native LotusD Implementation (#7125)

Browse Source 
* draft pass at a native comfy implementation of Lotus-D depth and normal est

* fix model_sampling kludges

* fix ruff

---------

Co-authored-by: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
This commit is contained in:
thot experiment
2025-03-21 11:04:15 -07:00
committed by GitHub
parent 0cf2274699
commit 83e839a89b
6 changed files with 74 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_model_advanced.py
View File

@@ -24,6 +24,10 @@ class X0(comfy.model_sampling.EPS):
def calculate_denoised(self, sigma, model_output, model_input):
return model_output
class Lotus(X0):
def calculate_input(self, sigma, noise):
return noise
class ModelSamplingDiscreteDistilled(comfy.model_sampling.ModelSamplingDiscrete):
original_timesteps = 50
@@ -56,7 +60,7 @@ class ModelSamplingDiscrete:
@classmethod
def INPUT_TYPES(s):
return {"required": { "model": ("MODEL",),
"sampling": (["eps", "v_prediction", "lcm", "x0"],),
"sampling": (["eps", "v_prediction", "lcm", "x0", "lotus"],),
"zsnr": ("BOOLEAN", {"default": False}),
}}
@@ -78,6 +82,8 @@ class ModelSamplingDiscrete:
sampling_base = ModelSamplingDiscreteDistilled
elif sampling == "x0":
sampling_type = X0
elif sampling == "lotus":
sampling_type = Lotus
class ModelSamplingAdvanced(sampling_base, sampling_type):
pass