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

Open

huvunvidia wants to merge 3 commits intoNVIDIA:devfrom

huvunvidia:huvu/diffusion_condition_embeddings

Commits on Apr 2, 2026

add conditions_embeddings
Huy Vu2
committed
fix unit test
Huy Vu2
committed

Commits on Apr 3, 2026

rename condition_embeddings to conditions_embeddings in HyperConnectionTransformerLayer

Huy Vu2
and
claude
committed