transformers

a40f1ac6 - Fix couples of issues from #36335 (#36453)

Fix couples of issues from #36335 (#36453) * fix * style * better allocation * fix * fix * style * revert disk * exit * style * return if nothing to cache * dtensor guard * fix regressiion * fix regression * fix * fix