Unpack DiT block output tuple in the non-checkpointed training branch - #86
Open
jeqcho wants to merge 1 commit into
Open
Unpack DiT block output tuple in the non-checkpointed training branch#86jeqcho wants to merge 1 commit into
jeqcho wants to merge 1 commit into
