Add conditions_embeddings argument to TransformerBlock, TransformerLayer for DiT (diffusion transformer)#4134

huvunvidia wants to merge 1 commit intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

huvunvidia:huvu/diffusion_condition_embeddings_mainhuvunvidia/Megatron-LM:huvu/diffusion_condition_embeddings_mainCopy head branch name to clipboard