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.