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

TRT decoder #45

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

TRT decoder #45

wants to merge 5 commits into from

Commits on Nov 8, 2024

  1. remove some einops usage

    @technillogue
    technillogue committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    3bf7029 View commit details
    Browse the repository at this point in the history

  2. formatting, initial changes, types, inlining, and initial attempt

    @technillogue
    technillogue committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    fd1546d View commit details
    Browse the repository at this point in the history

  3. use nn.Sequential to remove python control flow from autoencoder up/d… 

    …ownsampling
    @technillogue
    technillogue committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    9db707f View commit details
    Browse the repository at this point in the history

  4. try to keep the same state_dict structure 

    lazily create up_descending after state dict is already loaded, but only do it once
    @technillogue
    technillogue committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    29aa8c6 View commit details
    Browse the repository at this point in the history

  5. conditionally download and use decoder engine

    @technillogue
    technillogue committed Nov 8, 2024
    Configuration menu
    Copy the full SHA
    c982583 View commit details
    Browse the repository at this point in the history