Fix AutoEP ZeRO-1/2 universal conversion (#8198)