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

Use carried audio sampler state directly

Browse source
This commit is contained in:
Daniel Maddern 2026-08-13 21:23:18 +07:00
parent f5fe966a7e
commit b491589418
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/h3_blackwell_runtime/sampler.py
View file

@ -51,7 +51,7 @@ def _audio_sigma(video_sigma: torch.Tensor) -> torch.Tensor:
def sample_video_res_multistep(model, packer: H3PromptPacker, text: torch.Tensor, video: torch.Tensor, audio: torch.Tensor, *, steps: int = 12) -> torch.Tensor: def sample_video_res_multistep(model, packer: H3PromptPacker, text: torch.Tensor, video: torch.Tensor, audio: torch.Tensor, *, steps: int = 12) -> torch.Tensor:
"""Direct H3 beta/RES sampling with Comfy-equivalent joint AV carry semantics.""" """Direct H3 beta/RES sampling with Comfy-equivalent joint AV carry semantics."""
sigmas = beta_sigmas(steps, device=video.device) sigmas = beta_sigmas(steps, device=video.device)
audio_carried = audio * 4.0 audio_carried = audio
video_history = audio_history = None video_history = audio_history = None
video_history_sigma = audio_history_sigma = None video_history_sigma = audio_history_sigma = None
total_steps = len(sigmas) - 1 total_steps = len(sigmas) - 1