Smooth Things Come in Threes: A Diabatic Surrogate Model for Conical Intersection Optimization