the “projection” class_embed_type. Required when using the “projection” class_embed_type.



UNet2DConditionModel is a conditional 2D UNet model that takes in a noisy sample, conditional state, and a timestep
and returns sample shaped output.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the models (such as downloading or saving, etc.)

forward

<
source
>
(
sample