diffusion_models.models.openai_unet.TimestepEmbedSequential

class diffusion_models.models.openai_unet.TimestepEmbedSequential(*args, **kwargs)

Bases: Sequential, TimestepBlock

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

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)
forward(x, emb)	Apply the module to x given emb timestep embeddings.

forward(x, emb)

Apply the module to x given emb timestep embeddings.