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

Add new Relax function to the batched model for evaluating query tokens over multiple time steps in parallel #156

Merged
merged 7 commits into from
Jan 13, 2024
Merged

Add new Relax function to the batched model for evaluating query tokens over multiple time steps in parallel #156

merged 7 commits into from
Jan 13, 2024

Commits on Jan 11, 2024

  1. add new model for evaluating logits over multiple queries using KV cache

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    e0ef4c6 View commit details
    Browse the repository at this point in the history

  2. add test

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    4ccbb27 View commit details
    Browse the repository at this point in the history

  3. clean

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    f1314a5 View commit details
    Browse the repository at this point in the history

  4. Only the number of past tokens is needed

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    2bee022 View commit details
    Browse the repository at this point in the history

  5. fix build

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    756b09f View commit details
    Browse the repository at this point in the history

  6. fix

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    09ef5b3 View commit details
    Browse the repository at this point in the history

  7. correctly handle num_past_tokens > sliding_window case

    @masahi
    masahi committed Jan 11, 2024
    Configuration menu
    Copy the full SHA
    7b67ba4 View commit details
    Browse the repository at this point in the history