Custom DiT diffusion policy (from-scratch training loop)

b1c8c1d

verified

poxonit commited on 15 days ago