Skip to content

vllm_omni.diffusion.layers.activation ¶

Custom activation functions for diffusion models.

SiluAndMul ¶

Bases: CustomOp

SwiGLU activation dispatched by the current vLLM-Omni platform.

op `instance-attribute` ¶

op = torch.ops._C.silu_and_mul

forward_cuda ¶

forward_cuda(x: Tensor) -> Tensor

forward_native `staticmethod` ¶

forward_native(x: Tensor) -> Tensor

forward_npu ¶

forward_npu(x: Tensor) -> Tensor