Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add metadata format to save_file in FullModelHFCheckpointer #1126

Merged
merged 2 commits into from
Jun 27, 2024

Commits on Jun 26, 2024

  1. add metadata format to save_file in FullModelHFCheckpointer

    `AutoModel.xx` classes from hf's transformers do not work otherwise, and an error because the metadata is `None` here: https:/huggingface/transformers/blob/main/src/transformers/modeling_utils.py#L518-L522
    
    That said, I only have very basic knowledge of this so might be wrong here. I personally had to re-save the model files with metadata added for them to work with `AutoModel...` classes.
    man-shar authored Jun 26, 2024
    Configuration menu
    Copy the full SHA
    22929cf View commit details
    Browse the repository at this point in the history

Commits on Jun 27, 2024

  1. pre commit linting

    man-shar committed Jun 27, 2024
    Configuration menu
    Copy the full SHA
    3193d14 View commit details
    Browse the repository at this point in the history