Don't pass kwargs that aren't accepted by model_fn

Don't pass kwargs that aren't accepted by model_fn #2870