fix: honor attention backend for checkpoint inference

fix: honor attention backend for checkpoint inference #10561