Sparse tensor with support for both torchsparse and spconv backends. Parameters: - feats (torch.Tensor): Features of the sparse

Embeds spatial positions into vector representations.

A 3D subdivide block that can subdivide the sparse tensor. Args: channels: channels in the inputs and outputs. out_chan

Embeds scalar timesteps into vector representations.

A GroupNorm layer that converts to float32 before the forward pass.

Sparse Transformer cross-attention block (MSA + MCA + FFN) with adaptive layer norm conditioning.

Transformer cross-attention block (MSA + MCA + FFN) with adaptive layer norm conditioning.

A base class for pipelines.

Sparse Transformer block (MSA + FFN).

Pipeline for inferring Trellis image-to-3D models. Args: models (dict[str, nn.Module]): The models to use in the pipeline.

Basic manager for Trellis models, using ComfyUI's new model path.

Output class for Marigold monocular normals prediction pipeline. Args: prediction (`np.ndarray`, `torch.Tensor`): Predic

Pipeline for monocular normals estimation using the Marigold method: https://marigoldmonodepth.github.io. Pipeline for text-to-image generation u

A mixin class for samplers that apply classifier-free guidance.

Generate samples from a flow-matching model using Euler sampling with classifier-free guidance.

Generate samples from a flow-matching model using Euler sampling with classifier-free guidance and interval.

Generate samples from a flow-matching model using Euler sampling. Args: sigma_min: The minimum scale of noise in flow.

A mixin class for samplers that apply classifier-free guidance with interval.

Node to manage the loading of the TRELLIS model with lazy backend selection.

Sparse Transformer block (MSA + FFN) with adaptive layer norm conditioning.

Transformer block (MSA + FFN) with adaptive layer norm conditioning.

A base class for samplers.

Downsample a sparse tensor by a factor of `factor`. Implemented as average pooling.

A GroupNorm layer that converts to float32 before the forward pass.

A LayerNorm layer that converts to float32 before the forward pass.

Decoder for Sparse Structure (\mathcal{D}_S in the paper Sec. 3.3). Args: out_channels (int): Channels of the output.

Encoder for Sparse Structure (\mathcal{E}_S in the paper Sec. 3.3). Args: in_channels (int): Channels of the input.

Upsample a sparse tensor by a factor of `factor`. Implemented as nearest neighbor interpolation.

Sparse Transformer without output layers. Serve as the base class for encoder and decoder.

Sparse Transformer cross-attention block (MSA + MCA + FFN).

Upsample a sparse tensor by a factor of `factor`. Implemented as nearest neighbor interpolation.

Transformer block (MSA + FFN).

Transformer cross-attention block (MSA + MCA + FFN).