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.

Sign up for GitHub

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

Jump to bottom

fix: extend the unwrap_model function and save unwrapped model state dict instead of wrapped #29780

Closed
wants to merge 10 commits into from
Closed

fix: extend the unwrap_model function and save unwrapped model state dict instead of wrapped #29780

wants to merge 10 commits into from

Commits on Mar 21, 2024

  1. fix: extend the unwrap_model function to handle nested unwrapping in … 

    …children layers and save_unwrap_model state_dict instead of wrapped_model_state_dict
    @shub-kris
    shub-kris committed Mar 21, 2024
    Configuration menu
    Copy the full SHA
    62c86b2 View commit details
    Browse the repository at this point in the history

  2. fix: ruff formating

    @shub-kris
    shub-kris committed Mar 21, 2024
    Configuration menu
    Copy the full SHA
    44bec4f View commit details
    Browse the repository at this point in the history

Commits on Mar 26, 2024

  1. remove try except block to simplify unwrap_model()

    @shub-kris
    shub-kris committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    525ff47 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'huggingface:main' into fix/checkpointing-on-tpu-with-fs… 

    …dpv2
    @shub-kris
    shub-kris authored Mar 26, 2024
    Configuration menu
    Copy the full SHA
    41ea656 View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2024

  1. Merge branch 'huggingface:main' into fix/checkpointing-on-tpu-with-fs… 

    …dpv2
    @shub-kris
    shub-kris authored Mar 27, 2024
    Configuration menu
    Copy the full SHA
    9ed8b2d View commit details
    Browse the repository at this point in the history

  2. Add Tests for unwrap_model

    @shub-kris
    shub-kris committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    5ca4b2d View commit details
    Browse the repository at this point in the history

  3. fix ruff formatting, add require_torch

    @shub-kris
    shub-kris committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    456bba7 View commit details
    Browse the repository at this point in the history

  4. add missing import

    @shub-kris
    shub-kris committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    17398c0 View commit details
    Browse the repository at this point in the history

  5. fix import order

    @shub-kris
    shub-kris committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    f39af00 View commit details
    Browse the repository at this point in the history

  6. remove type-checking from unwrap_model_old()

    @shub-kris
    shub-kris committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    1e969ee View commit details
    Browse the repository at this point in the history