Methodpatch(
cls,
*,
model,
seed,
tile_size,
swap_size,
max_depth
py/nodes/hyperTile.py:186
Functionsageattn_sampler(
config: SageAttnOptions,
model: object,
x: torch.Tensor,
sigmas: torch.Tensor,
# *,
py/nodes/sageAttention.py:385
Functionwavelet_blend(
a: tuple,
b: tuple,
*,
yl_factor: torch.Tensor | float,
blend_function: Callable,
yh
py/wavelet_functions.py:228