vllm_omni.diffusion.layers.fourier ¶

GaussianFourierProjection ¶

Bases: Module

Shared Gaussian Fourier features with optional trainable frequencies.

embedding_size = int(embedding_size)

in_features = int(in_features)

weight = nn.Parameter(weight, requires_grad=trainable)

forward(x: Tensor) -> Tensor