A sequential module that passes timestep embeddings to the children that support it as an extra input.

dictionary with tensor keys and tensor value warning : this is not actual dictionary that uses hash

dictionary with tensor keys and tensor value warning : this is not actual dictionary that uses hash

InceptionC block patched for FID computation

InceptionC block patched for FID computation

A residual block that can optionally change the number of channels. :param channels: the number of input channels. :param emb_channels:

An attention block that allows spatial positions to attend to each other. Originally ported from here, but adapted to the N-d case. http

A downsampling layer with an optional convolution. :param channels: channels in the inputs and outputs. :param use_conv: a bool determin

InceptionA block patched for FID computation

InceptionA block patched for FID computation

`LSUN <https://www.yf.io/p/lsun>`_ dataset. Args: root (string): Root directory for the database files. classes (string or l

`LSUN <https://www.yf.io/p/lsun>`_ dataset. Args: root (string): Root directory for the database files. classes (string or l

An upsampling layer with an optional convolution. :param channels: channels in the inputs and outputs. :param use_conv: a bool determini

Utilities for training and sampling DLPMa and LIM diffusion models. Ported directly from here, and then adapted over time to further experime

Pretrained InceptionV3 network returning feature maps

Pretrained InceptionV3 network returning feature maps

NCSN++ model

First InceptionE block patched for FID computation

First InceptionE block patched for FID computation

Second InceptionE block patched for FID computation

Second InceptionE block patched for FID computation

A module which performs QKV attention.

Channel-wise self-attention block.

Channel-wise self-attention block.

Channel-wise self-attention block. Modified from DDPM.

Channel-wise self-attention block. Modified from DDPM.

Combine information from skip connections.

Combine information from skip connections.