abeeha123 opened a new pull request, #18503: URL: https://github.com/apache/tvm/pull/18503
Fixes #18357 — KeyError in the Relax PyTorch frontend when loading an HuggingFace BERT model exported with torch. export. The issue occurs because non-persistent buffers, such as position_ids and token_type_ids, are not included in the state_dict, leading to missing-buffer errors. This PR adds support for extra non-persistent buffers by introducing an extra_buffers lookup and extending buffer-resolution logic to handle shapes and dtypes correctly. With this fix, HuggingFace BERT models exported via torch. export no longer fails with missing-buffer KeyErrors. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
