🐛 Describe the bug

The function was meant to align dynamic_shapes keys to model signature, since torch.export.export does not support customized input_names. However, model.signature could be the same length as input_names accidentally. For example, input_names includes all kv_cache names, and LLM models usually come with a lot of Optional inputs, while kv_cache names only counts one because it's a list.

Versions

Nightly