adapted code from https://github.com/comfyanonymous/ComfyUI/pull/3370 TODO: check if virtual scheduler tcd is needed for using sampling_base ModelSamplingDiscreteDistilled or if it's better to use sgm_uniform directly without patching