Apply torch.compile (inductor) to the Anima DiT one transformer block at a time. Faster compile and fewer graph breaks than whole-model compile. — ComfyUI node…

Anima Block Compile for ComfyUI: a one-purpose node that runs torch.compile on the Anima DiT per transformer block (diffusion_model.blocks.{i}) instead of the whole model. Faster compile, fewer graph breaks. A thin, Anima-named wrapper over ComfyUI core's set_torch_compile_wrapper.

Anima Block Compile

model

Apply torch.compile (inductor) to the Anima DiT one transformer block at a time. Faster compile and fewer graph breaks than whole-model compile.

Anima Block Compile

Inputs (1)

Outputs (1)