diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-17 12:16:40 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-17 12:16:40 +0100 |
| commit | c3176f0dfb8ace213e4a499918da21bc5712a107 (patch) | |
| tree | dea70edb1fe0f005e2f3c354a9d48a52ddc41ea0 /candle-nn/examples/basic_optimizer.rs | |
| parent | 03be33eea482accbcf4c547728c2db7e24b7ebb0 (diff) | |
| download | candle-c3176f0dfb8ace213e4a499918da21bc5712a107.tar.gz candle-c3176f0dfb8ace213e4a499918da21bc5712a107.tar.bz2 candle-c3176f0dfb8ace213e4a499918da21bc5712a107.zip | |
Flash-attention support in stable diffusion (#487)
* Add flash-attention for the stable-diffusion example.
* Change the dtype.
* Silly fix.
* Another fix.
* Revert the dtype back to the query dtype after apply flash-attn.
Diffstat (limited to 'candle-nn/examples/basic_optimizer.rs')
0 files changed, 0 insertions, 0 deletions