fixed bug where pred_mode was not making it to the model params