transformers

fix num_assistant_tokens with heuristic schedule

#28759

Merged