add model to rollout args