Fix: num_logits_to_keep in composite models
#33168
Merged
num_logits_to_keep in composite models
#33168