candle/candle-wasm-tests
Czxck001 ca7cf5cb3b
Add Stable Diffusion 3 Example (#2558)
* 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

* fix typos

* expose cfg_scale and time_shift as options

* Replace the sample image with JPG version. Change image output format accordingly.

* make meaningful error messages

* remove the tail-end assignment in sd3_vae_vb_rename

* remove the CUDA requirement

* use default_value in clap args

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

* resolve clippy errors and warnings

* use default_value_t

* Pin the web-sys dependency.

* Clippy fix.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>
2024-10-13 22:08:40 +02:00
..
src Simd128 vec-dot for q4_0. (#974) 2023-09-27 14:15:30 +01:00
tests Add Stable Diffusion 3 Example (#2558) 2024-10-13 22:08:40 +02:00
Cargo.toml Simplifying our internal cargo dependencies. (#1529) 2024-01-07 12:04:14 +01:00
README.md Simd128 vec-dot for q4_0. (#974) 2023-09-27 14:15:30 +01:00
webdriver.json Simd128 vec-dot for q4_0. (#974) 2023-09-27 14:15:30 +01:00

README.md

Run the tests with:

RUST_LOG=wasm_bindgen_test_runner wasm-pack test --chrome --headless

Or:

wasm-pack test --chrome

If you get an "invalid session id" failure in headless mode, check that logs and it may well be that your ChromeDriver is not at the same version as your browser.