huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 04:00:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the MMDiT model of Stable Diffusion 3 (#2397)

Browse Source 
* add mmdit of stable diffusion 3

lint

add comments

* correct a misplaced comment

* fix cargo fmt

* fix clippy error

* use bail! instead of assert!

* use get_on_dim in splitting qkv
This commit is contained in:
Czxck001
2024-08-05 10:26:15 -07:00
committed by GitHub
parent 500c9f2882
commit dfdce2b602
6 changed files with 763 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-transformers/src/models/mmdit/mod.rs Normal file
View File

@ -0,0 +1,4 @@
pub mod blocks;
pub mod embedding;
pub mod model;
pub mod projections;