From 2ef5d002709855d4a3a70fff4791ea3ad27d6fc8 Mon Sep 17 00:00:00 2001 From: Daniel Maddern Date: Thu, 13 Aug 2026 22:39:47 +0700 Subject: [PATCH] Use Comfy model sigma in free-run comparator --- tools/compare_fl2va_free_run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/compare_fl2va_free_run.py b/tools/compare_fl2va_free_run.py index bd82de5..ef39a5f 100644 --- a/tools/compare_fl2va_free_run.py +++ b/tools/compare_fl2va_free_run.py @@ -7,7 +7,7 @@ import torch from h3_blackwell_runtime.checkpoint import H3Checkpoint from h3_blackwell_runtime.denoiser import H3PackedDenoiser from h3_blackwell_runtime.packing import H3PromptPacker, unpatchify_video -from h3_blackwell_runtime.sampler import _audio_sigma, _unpack_audio, res_multistep_update +from h3_blackwell_runtime.sampler import _audio_sigma, _model_sigma, _unpack_audio, res_multistep_update from h3_blackwell_runtime.t2v import random_av_latents @@ -51,7 +51,7 @@ for index, sigma in enumerate(sigmas[:-1]): text, video, audio_carried.to(torch.bfloat16) * carry, - float(sigma), + _model_sigma(sigma), model_timesteps, ) hidden_delta = (hidden.float() - h3_input["hidden"].to("cuda").float()).abs()