diffusion_models.models.openai_unet.TimestepBlock

class diffusion_models.models.openai_unet.TimestepBlock(*args, **kwargs)[source]

Bases: Module

Any module where forward() takes timestep embeddings as a second argument.

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)

forward(x, emb)

Apply the module to x given emb timestep embeddings.

abstract forward(x, emb)[source]

Apply the module to x given emb timestep embeddings.