Skip to content

vllm_omni.diffusion.models.hunyuan_image3.layers.common

Modules:

Name Description
utils

Small platform-agnostic helpers for UNet-style blocks (adapted from diffusers).

conv_nd

conv_nd(dims, *args, **kwargs)

Create a 1D, 2D, or 3D convolution module.

linear

linear(*args, **kwargs)

Create a linear module.

normalization

normalization(channels, **kwargs)

Make a standard normalization layer. :param channels: number of input channels. :return: a nn.Module for normalization.

zero_module

zero_module(module)

Zero out the parameters of a module and return it.