Function _wavelet_reconstruct

diffsynth/pipelines/flashvsr_tiny_long.py:78–81

(content: torch.Tensor, style: torch.Tensor, levels: int = 5)

def _wavelet_reconstruct(content: torch.Tensor, style: torch.Tensor, levels: int = 5) -> torch.Tensor:
    c_high, _ = _wavelet_decompose(content, levels=levels)
    _, s_low = _wavelet_decompose(style, levels=levels)
    return c_high + s_low


# -----------------------------