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

Fixes Imagen sampling example and updates container #868

Merged
merged 13 commits into from
Aug 26, 2024
Merged

Fixes Imagen sampling example and updates container #868

merged 13 commits into from
Aug 26, 2024

Commits on Jul 22, 2024

  1. Fix imagen loop when prompt_ct is a multiple of 

    batch_size//gen_per_prompt
    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    cc4290b View commit details
    Browse the repository at this point in the history

  2. Add a comment explaining the invocation of the sampling script for 

    imagen
    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    c35a14f View commit details
    Browse the repository at this point in the history

  3. Update imagen image with this fix: 

    ```
docker buildx build --push -t ghcr.io/nvidia/t5x:imagen-2023-10-02.v2 . -f - <<EOF
FROM ghcr.io/nvidia/t5x:imagen-2023-10-02
COPY rosetta/rosetta/projects/imagen/imagen_pipe.py /opt/rosetta/rosetta/projects/imagen/imagen_pipe.py
EOF
```
    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    c0d4885 View commit details
    Browse the repository at this point in the history

  4. parametrize batch size for imagen

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    4caed81 View commit details
    Browse the repository at this point in the history

  5. update documentation with new keys and values

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    86511b2 View commit details
    Browse the repository at this point in the history

  6. add a timer 

    docker buildx build --push -t ghcr.io/nvidia/t5x:imagen-2023-10-02.v2 .
-f - <<EOF
FROM ghcr.io/nvidia/t5x:imagen-2023-10-02
COPY rosetta/rosetta/projects/imagen/README.md
/opt/rosetta/rosetta/projects/imagen/README.md
COPY rosetta/rosetta/projects/imagen/imagen_pipe.py
/opt/rosetta/rosetta/projects/imagen/imagen_pipe.py
COPY rosetta/rosetta/projects/imagen/scripts/sample_imagen_1024.sh
/opt/rosetta/rosetta/projects/imagen/scripts/sample_imagen_1024.sh
COPY rosetta/rosetta/projects/imagen/scripts/sample_imagen_256.sh
/opt/rosetta/rosetta/projects/imagen/scripts/sample_imagen_256.sh
EOF
    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    7c55439 View commit details
    Browse the repository at this point in the history

  7. typo

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    0c235e3 View commit details
    Browse the repository at this point in the history

  8. better logging

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    3cf8c94 View commit details
    Browse the repository at this point in the history

  9. update readme on the divisible by gpu and add MAX_FILENAME_LENGTH

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    cc753e1 View commit details
    Browse the repository at this point in the history

  10. typo in perf calc

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    9940b9c View commit details
    Browse the repository at this point in the history

  11. reset time

    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    6baba82 View commit details
    Browse the repository at this point in the history

  12. few other missing imagen tags updated 

    Signed-off-by: Terry Kong <[email protected]>
    @terrykong
    terrykong committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    c17c496 View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2024

  1. fixes for 2b

    @terrykong
    terrykong committed Aug 2, 2024
    Configuration menu
    Copy the full SHA
    24ebd1d View commit details
    Browse the repository at this point in the history