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 Stable Diffusion 3 Example #2558

Merged
merged 13 commits into from
Oct 13, 2024
Merged

Add Stable Diffusion 3 Example #2558

merged 13 commits into from
Oct 13, 2024

Commits on Oct 12, 2024

  1. Add stable diffusion 3 example 

    Add get_qkv_linear to handle different dimensionality in linears

Add stable diffusion 3 example

Add use_quant_conv and use_post_quant_conv for vae in stable diffusion

adapt existing AutoEncoderKLConfig to the change

add forward_until_encoder_layer to ClipTextTransformer

rename sd3 config to sd3_medium in mmdit; minor clean-up

Enable flash-attn for mmdit impl when the feature is enabled.

Add sd3 example codebase

add document

crediting references

pass the cargo fmt test

pass the clippy test
    @Czxck001
    Czxck001 committed Oct 12, 2024
    Configuration menu
    Copy the full SHA
    572b4cb View commit details
    Browse the repository at this point in the history

Commits on Oct 13, 2024

  1. fix typos

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    443824b View commit details
    Browse the repository at this point in the history

  2. expose cfg_scale and time_shift as options

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    64c9181 View commit details
    Browse the repository at this point in the history

  3. Replace the sample image with JPG version. Change image output format… 

    … accordingly.
    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    84e6463 View commit details
    Browse the repository at this point in the history

  4. make meaningful error messages

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    c56c98b View commit details
    Browse the repository at this point in the history

  5. remove the tail-end assignment in sd3_vae_vb_rename

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    24691ad View commit details
    Browse the repository at this point in the history

  6. remove the CUDA requirement

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    b8f0923 View commit details
    Browse the repository at this point in the history

  7. use default_value in clap args

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    457e188 View commit details
    Browse the repository at this point in the history

  8. add use_flash_attn to turn on/off flash-attn for MMDiT at runtime

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    50da3fe View commit details
    Browse the repository at this point in the history

  9. resolve clippy errors and warnings

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    e37a1f6 View commit details
    Browse the repository at this point in the history

  10. use default_value_t

    @Czxck001
    Czxck001 committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    8f1339e View commit details
    Browse the repository at this point in the history

  11. Pin the web-sys dependency.

    @LaurentMazare
    LaurentMazare committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    1fc6482 View commit details
    Browse the repository at this point in the history

  12. Clippy fix.

    @LaurentMazare
    LaurentMazare committed Oct 13, 2024
    Configuration menu
    Copy the full SHA
    a2e4ecd View commit details
    Browse the repository at this point in the history