Model with reduce_sum takes too long

I guess not always.