[feat]: upload vae_encoder
#2
by
huisu - opened
- .gitattributes +4 -0
- vae_encoder/config.json +31 -0
- vae_encoder/dd/cache/NhwcConv_0-encoderconv_inConv_meta.json +0 -0
- vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.ctrlpkt +3 -0
- vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.fconst +3 -0
- vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.state +3 -0
- vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.super +3 -0
- vae_encoder/dd/onnx_report.txt +42 -0
- vae_encoder/dd/replaced.onnx +3 -0
.gitattributes
CHANGED
|
@@ -41,3 +41,7 @@ vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.ctrlpkt
|
|
| 41 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.fconst filter=lfs diff=lfs merge=lfs -text
|
| 42 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.state filter=lfs diff=lfs merge=lfs -text
|
| 43 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.super filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.fconst filter=lfs diff=lfs merge=lfs -text
|
| 42 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.state filter=lfs diff=lfs merge=lfs -text
|
| 43 |
vae_decoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-post_quant_convConv.super filter=lfs diff=lfs merge=lfs -text
|
| 44 |
+
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.ctrlpkt filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.fconst filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.state filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.super filter=lfs diff=lfs merge=lfs -text
|
vae_encoder/config.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "AutoencoderKL",
|
| 3 |
+
"_diffusers_version": "0.24.0.dev0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"block_out_channels": [
|
| 6 |
+
128,
|
| 7 |
+
256,
|
| 8 |
+
512,
|
| 9 |
+
512
|
| 10 |
+
],
|
| 11 |
+
"down_block_types": [
|
| 12 |
+
"DownEncoderBlock2D",
|
| 13 |
+
"DownEncoderBlock2D",
|
| 14 |
+
"DownEncoderBlock2D",
|
| 15 |
+
"DownEncoderBlock2D"
|
| 16 |
+
],
|
| 17 |
+
"force_upcast": true,
|
| 18 |
+
"in_channels": 3,
|
| 19 |
+
"latent_channels": 4,
|
| 20 |
+
"layers_per_block": 2,
|
| 21 |
+
"norm_num_groups": 32,
|
| 22 |
+
"out_channels": 3,
|
| 23 |
+
"sample_size": 1024,
|
| 24 |
+
"scaling_factor": 0.13025,
|
| 25 |
+
"up_block_types": [
|
| 26 |
+
"UpDecoderBlock2D",
|
| 27 |
+
"UpDecoderBlock2D",
|
| 28 |
+
"UpDecoderBlock2D",
|
| 29 |
+
"UpDecoderBlock2D"
|
| 30 |
+
]
|
| 31 |
+
}
|
vae_encoder/dd/cache/NhwcConv_0-encoderconv_inConv_meta.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.ctrlpkt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2e1a5c24f90e1502dbfc5f79348dcb2f618bb1cb0bf9b6b88ceec2bcbd8d0241
|
| 3 |
+
size 29858688
|
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.fconst
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:39f4c3ac20324cc894cc5333ee3b21209bc92cfbb7cfd26ec7964e5f94330767
|
| 3 |
+
size 42147856
|
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.state
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cbb30ef89acf22555554aaee0a8a8b1fd7c08fc8618b3f64c11b791bfa172207
|
| 3 |
+
size 6466405
|
vae_encoder/dd/dd_metastate_Sd15_Decoder_NhwcConv_0-encoderconv_inConv.super
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:230fe28efa807448cf48c4d03c05becc580249daabc6b2a7120f35cdbcca2414
|
| 3 |
+
size 772640
|
vae_encoder/dd/onnx_report.txt
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
DynamicDispatch Offload - not offloaded
|
| 2 |
+
+-------------------+-------+------------------------+------------------------+
|
| 3 |
+
| Op Type | Count | Inputs | Outputs |
|
| 4 |
+
+===================+=======+========================+========================+
|
| 5 |
+
| CastAvx | 1 | [1,1024,1024,4] - | [1,1024,1024,4] - |
|
| 6 |
+
| | | FLOAT | BFLOAT16 |
|
| 7 |
+
| CastAvx | 1 | [1,128,128,8] - | [1,128,128,8] - FLOAT |
|
| 8 |
+
| | | BFLOAT16 | |
|
| 9 |
+
| Pad | 1 | [1,3,1024,1024] - | [1,4,1024,1024] - |
|
| 10 |
+
| | | FLOAT | FLOAT |
|
| 11 |
+
| | | [8] - INT64 | |
|
| 12 |
+
| | | [1] - FLOAT | |
|
| 13 |
+
| Slice | 1 | [1,8,128,128] - FLOAT | [1,4,128,128] - FLOAT |
|
| 14 |
+
| | | [1] - INT64 | |
|
| 15 |
+
| | | [1] - INT64 | |
|
| 16 |
+
| | | [1] - INT64 | |
|
| 17 |
+
| Transpose | 1 | [1,4,1024,1024] - | [1,1024,1024,4] - |
|
| 18 |
+
| | | FLOAT | FLOAT |
|
| 19 |
+
| Transpose | 1 | [1,128,128,8] - FLOAT | [1,8,128,128] - FLOAT |
|
| 20 |
+
+-------------------+-------+------------------------+------------------------+
|
| 21 |
+
| Not offloaded sum | 6 | | |
|
| 22 |
+
+-------------------+-------+------------------------+------------------------+
|
| 23 |
+
DynamicDispatch Offload - offloaded
|
| 24 |
+
+--------------------+-------------+--------------------+---------------------+
|
| 25 |
+
| Op Type | Count | Inputs | Outputs |
|
| 26 |
+
+====================+=============+====================+=====================+
|
| 27 |
+
| DynamicDispatch | 1 | [1,1024,1024,4] - | [1,128,128,8] - |
|
| 28 |
+
| | | BFLOAT16 | BFLOAT16 |
|
| 29 |
+
+--------------------+-------------+--------------------+---------------------+
|
| 30 |
+
| Offloaded sum | 1 | | |
|
| 31 |
+
+--------------------+-------------+--------------------+---------------------+
|
| 32 |
+
| Offloaded Op Types | SDAdd | | |
|
| 33 |
+
| | SDMHA_VAE | | |
|
| 34 |
+
| | SDConv | | |
|
| 35 |
+
| | SDGroupNorm | | |
|
| 36 |
+
| | SDGemm | | |
|
| 37 |
+
+--------------------+-------------+--------------------+---------------------+
|
| 38 |
+
| Offloaded sum (dd | 66 | | |
|
| 39 |
+
| fusion) | | | |
|
| 40 |
+
| Offload Ratio (dd | 91.67% | | |
|
| 41 |
+
| fusion) | | | |
|
| 42 |
+
+--------------------+-------------+--------------------+---------------------+
|
vae_encoder/dd/replaced.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0ec267beaa5e8681a2494a4d64adb0ad15a9ed8274926ff171124e407ca32acc
|
| 3 |
+
size 40934
|