accelerate

beed693e - Prepare TP fix (#3945)

Prepare TP fix (#3945) * fix(tp): skip tp preparation for models not using dtensors * fix: move block right after schedulers preparation