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

Advanced DiT Training #11226

Closed
wants to merge 1 commit into from
Closed

Advanced DiT Training #11226

wants to merge 1 commit into from

Commits on Nov 9, 2024

  1. new training features 

    Signed-off-by: Zeeshan Patel <[email protected]>

packed sequence support

Signed-off-by: Zeeshan Patel <[email protected]>

fixed pylint issues

Signed-off-by: Zeeshan Patel <[email protected]>

add docstrings + remove unused import

Signed-off-by: Zeeshan Patel <[email protected]>

Apply isort and black reformatting

Signed-off-by: zpx01 <[email protected]>
Signed-off-by: Zeeshan Patel <[email protected]>

added some missing args

Signed-off-by: Zeeshan Patel <[email protected]>

reformat

Signed-off-by: Zeeshan Patel <[email protected]>

moving fps embedder to gpu

Signed-off-by: Zeeshan Patel <[email protected]>

moving fps tensor  to gpu

Signed-off-by: Zeeshan Patel <[email protected]>

updated models

Signed-off-by: Zeeshan Patel <[email protected]>

added missing import

Signed-off-by: Zeeshan Patel <[email protected]>

added new configs + bug fixes

Signed-off-by: Zeeshan Patel <[email protected]>

seq len fix

Signed-off-by: Zeeshan Patel <[email protected]>

Apply isort and black reformatting

Signed-off-by: zpx01 <[email protected]>
Signed-off-by: Zeeshan Patel <[email protected]>

added nsys callback

Signed-off-by: Zeeshan Patel <[email protected]>

Apply isort and black reformatting

Signed-off-by: zpx01 <[email protected]>
Signed-off-by: Zeeshan Patel <[email protected]>

isort fix

Signed-off-by: Zeeshan Patel <[email protected]>
    Zeeshan Patel committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    3805601 View commit details
    Browse the repository at this point in the history