It's pretty easy to do... you just set the wall to be the total height you want and make sure the story settings are set to "Automatic" and "On Relevant Stories"
I tend to make mine profiled walls with consistent sheathing going all the way up, stud top plates, a gap for the floor truss, then the next floor bottom plate.
Tom Waltz