Class LTXVNormalizingSampler

easy_samplers.py:916–1051

@comfy_node(description="LTXV Normalizing Sampler")
class LTXVNormalizingSampler(io.ComfyNode):
    @classmethod
    def define_schema(cls):
        return io.Schema(
            node_id="LTXVNormalizingSampler",
            category="utility",
            inputs=[
                io.Noise.Input("noise"),
                io.Guider.Input("guider"),
                io.Sampler.Input("sampler"),
                io.Sigmas.Input("sigmas"),
                io.Latent.Input("latent_image"),
                io.String.Input(
                    "video_normalization_factors", default="1,1,1,1,1,1,1,1"
                ),
                io.String.Input(
                    "audio_normalization_factors", default="1,1,0.25,1,1,0.25,1,1"
                ),
            ],
            outputs=[
                io.Latent.Output(display_name="denoised_output"),
            ],
        )

    @classmethod
    def execute(
        cls,
        noise,
        guider,
        sampler,
        sigmas,
        latent_image,
        video_normalization_factors,
        audio_normalization_factors,
    ) -> io.NodeOutput:

        if (
            guider.model_patcher.model.diffusion_model.__class__.__name__
            != "LTXAVModel"
        ):
            raise ValueError()

        ltxav: LTXAVModel = guider.model_patcher.model.diffusion_model

        video_normalization_factors = video_normalization_factors.split(",")
        audio_normalization_factors = audio_normalization_factors.split(",")
        video_normalization_factors = [
            float(factor) for factor in video_normalization_factors
        ]
        audio_normalization_factors = [
            float(factor) for factor in audio_normalization_factors
        ]

        # Extend normalization factors to match the length of sigmas
        sigmas_len = len(sigmas) - 1
        if len(video_normalization_factors) < sigmas_len:
            if len(video_normalization_factors) > 0:
                video_normalization_factors.extend(