vllm_omni.diffusion.utils.flow_matching ¶ SinusoidalPositionalEncoding ¶ Bases: Module embedding_dim instance-attribute ¶ embedding_dim = embedding_dim forward ¶ forward(timesteps: Tensor) -> Tensor safe_linalg_solve ¶ safe_linalg_solve(matrix: Tensor, rhs: Tensor) -> Tensor swish ¶ swish(x: Tensor) -> Tensor