daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use Comfy model sigma in free-run comparator

Browse source
This commit is contained in:
Daniel Maddern 2026-08-13 22:39:47 +07:00
parent 4018fbea15
commit 2ef5d00270
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tools/compare_fl2va_free_run.py
View file

@ -7,7 +7,7 @@ import torch
from h3_blackwell_runtime.checkpoint import H3Checkpoint from h3_blackwell_runtime.checkpoint import H3Checkpoint
from h3_blackwell_runtime.denoiser import H3PackedDenoiser from h3_blackwell_runtime.denoiser import H3PackedDenoiser
from h3_blackwell_runtime.packing import H3PromptPacker, unpatchify_video 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 from h3_blackwell_runtime.t2v import random_av_latents
@ -51,7 +51,7 @@ for index, sigma in enumerate(sigmas[:-1]):
text, text,
video, video,
audio_carried.to(torch.bfloat16) * carry, audio_carried.to(torch.bfloat16) * carry,
float(sigma), _model_sigma(sigma),
model_timesteps, model_timesteps,
) )
hidden_delta = (hidden.float() - h3_input["hidden"].to("cuda").float()).abs() hidden_delta = (hidden.float() - h3_input["hidden"].to("cuda").float()).abs()