Increase output precision on input data for small model