more fixes to get 1.2 behavior back in log_prob templated functions