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

TF generate refactor - Sample #15793

Merged
merged 11 commits into from Mar 2, 2022
Merged

TF generate refactor - Sample #15793

merged 11 commits into from Mar 2, 2022

Commits on Mar 2, 2022

  1. Add TFTopPLogitsWarper (and other minor changes)

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    e2280db View commit details
    Browse the repository at this point in the history

  2. Add sample method

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    1db4e89 View commit details
    Browse the repository at this point in the history

  3. push hacky past later in generate

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    7a2edb9 View commit details
    Browse the repository at this point in the history

  4. make fixup

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    abab083 View commit details
    Browse the repository at this point in the history

  5. fix (most) tests

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    95fbce6 View commit details
    Browse the repository at this point in the history

  6. Redo TopK to avoid corner case

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    29fbcc5 View commit details
    Browse the repository at this point in the history

  7. Better comments

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    d0c28c6 View commit details
    Browse the repository at this point in the history

  8. add tests for TF logit wrapers

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    bffbac5 View commit details
    Browse the repository at this point in the history

  9. Update src/transformers/generation_tf_logits_process.py 

    Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
    @gante @Rocketknight1
    gante and Rocketknight1 committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    9daa6ec View commit details
    Browse the repository at this point in the history

  10. Update src/transformers/generation_tf_logits_process.py 

    Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
    @gante @Rocketknight1
    gante and Rocketknight1 committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    04e4738 View commit details
    Browse the repository at this point in the history

  11. TF generate sample tests now run on CPU

    @gante
    gante committed Mar 2, 2022
    Configuration menu
    Copy the full SHA
    8b94e41 View commit details
    Browse the repository at this point in the history