The parameterization of inertia–gravity waves (IGWs) is of considerable importance in general circulation models. Among the challenging issues faced in studies concerned with parameterization of IGWs is the estimation of diabatic forcing in a way independent of the physics parameterization schemes, in particular, convection. The requirement is to estimate the diabatic heating associated with balanced motion. This can be done by comparing estimates of balanced vertical motion with and without diabatic effects. The omega equation provides the natural method of estimating balanced vertical motion without diabatic effects, and several methods for including diabatic effects are compared. To this end, the assumption of spatial-scale separation between IGWs and balanced flows is combined with a suitable form of the balanced omega equation. To test the methods constructed for estimating diabatic heating, an idealized numerical simulation of the moist baroclinic waves is performed using the Weather Research and Forecasting (WRF) Model in a channel on the f plane. In overall agreement with the diabatic heating of the WRF Model, in the omega-equation-based estimates, the maxima of heating appear in the warm sector of the baroclinic wave and in the exit region of the upper-level jet. The omega-equation-based method with spatial smoothing for estimating balanced vertical motion is thus presented as the proper way to evaluate diabatic forcing for parameterization of IGWs.