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

do NOT review - offloading for other recipes #1578

Draft
wants to merge 10 commits into
base: main
Choose a base branch
from
Draft

do NOT review - offloading for other recipes #1578

wants to merge 10 commits into from

Commits on Sep 11, 2024

  1. Add activations offloading API

    @janeyx99
    janeyx99 committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    c4f0366 View commit details
    Browse the repository at this point in the history

  2. Address Felipe's prev review i missed, try docs again

    @janeyx99
    janeyx99 committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    f1178c7 View commit details
    Browse the repository at this point in the history

  3. Fix the docs issue hehehehhehehehe, and use_streams=False

    @janeyx99
    janeyx99 committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    c937396 View commit details
    Browse the repository at this point in the history

  4. use_streams False, update docs, make hack hackier

    @janeyx99
    janeyx99 committed Sep 11, 2024
    Configuration menu
    Copy the full SHA
    9eea976 View commit details
    Browse the repository at this point in the history

Commits on Sep 13, 2024

  1. Merge branch 'main' into activations-offloading-streams

    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    c51db85 View commit details
    Browse the repository at this point in the history

  2. add lora distributed

    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    12812dd View commit details
    Browse the repository at this point in the history

  3. add full distributed

    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    b7c0d30 View commit details
    Browse the repository at this point in the history

  4. add full single

    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    41a14d5 View commit details
    Browse the repository at this point in the history

  5. add contextlib

    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    1ae7a6d View commit details
    Browse the repository at this point in the history

  6. Merge branch 'main' of github.com:pytorch/torchtune into add_offloadi… 

    …ng_other_recipes
    Felipe Mello committed Sep 13, 2024
    Configuration menu
    Copy the full SHA
    7aa7e49 View commit details
    Browse the repository at this point in the history