Towards Generalization of Graph Neural Networks for AC Optimal Power Flow