mirror of
https://github.com/immich-app/ml-models.git
synced 2026-09-30 13:22:55 +08:00
60e028dee6
* feat: add ppocr to exporter * ocr: unpack scrfd heads * ocr: affine fold Co-authored-by: todorangrg <todorangrg@gmail.com> * rknn dynamic inputs * fp16 outputs * tiled transpose * refactor * target multiple aspect ratios * update cli * add 16:9 * better ci fixture integration * ci bug fixes * check cancellation * chore: bless ci fixtures * tweak runners * add metadata to rknn binaries * canvas->dim * fix fixture trigger * update onnxscript locals * chore: bless ci fixtures * download unconditionally * simplify fix-fixtures * use huge runner for server model * run compilation in subprocess * try/catch recv * remove 16/9 * fix detection mac utilization * chore: bless ci fixtures * conditional rknn compile * add rknn config to render --------- Co-authored-by: todorangrg <todorangrg@gmail.com> Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com>
125 lines
4.0 KiB
Plaintext
125 lines
4.0 KiB
Plaintext
CPUExecutionProvider
|
|
-23 Add
|
|
-23 LayerNormalization axis=-1 epsilon=9.999999974752427e-07 stash_type=1
|
|
+23 SkipLayerNormalization epsilon=9.999999974752427e-07
|
|
168 nodes, weights 1ea1bf8923bd
|
|
FuseSkipLayerNorm x23 6d24dab3
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x141 08fea7e7
|
|
|
|
CUDAExecutionProvider
|
|
-23 Add
|
|
-23 LayerNormalization axis=-1 epsilon=9.999999974752427e-07 stash_type=1
|
|
+23 SkipLayerNormalization epsilon=9.999999974752427e-07
|
|
168 nodes, weights 1ea1bf8923bd
|
|
FuseSkipLayerNorm x23 6d24dab3
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x141 08fea7e7
|
|
|
|
CoreMLExecutionProvider
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+24 MatMul
|
|
+13 Mul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
-1 ReduceL2 keepdims=1 noop_with_empty_axes=0
|
|
+1 ReduceSum keepdims=1
|
|
+1 Sqrt
|
|
337 nodes, weights 287268b0361f
|
|
DecomposeAttention x156 f5d0980b
|
|
DecomposeReduceL2 x3 30350c72
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x174 d85bdc66
|
|
|
|
MIGraphXExecutionProvider
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+24 MatMul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Mul
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
335 nodes, weights 287268b0361f
|
|
DecomposeAttention x156 f5d0980b
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x175 ea6c9832
|
|
|
|
NvTensorRTRTXExecutionProvider
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+24 MatMul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Mul
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
335 nodes, weights 287268b0361f
|
|
DecomposeAttention x156 f5d0980b
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x175 ea6c9832
|
|
|
|
OpenVINOExecutionProvider
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+24 MatMul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Mul
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
335 nodes, weights 287268b0361f
|
|
DecomposeAttention x156 f5d0980b
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x175 ea6c9832
|
|
|
|
RKNPU static
|
|
+84 MatMul
|
|
+72 Add
|
|
+72 Slice
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+36 Mul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Div
|
|
+12 Erf
|
|
-12 Gelu approximate=none
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
575 nodes, weights d979523f1944
|
|
contract {"dims":[{}]}
|
|
opset 19
|
|
DecomposeAttention x156 f5d0980b
|
|
DecomposeGelu x60 a6662a53
|
|
SplitLargeReduction x204 1e2c9a0f
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x151 508d039a
|
|
|
|
TensorrtExecutionProvider
|
|
+48 Reshape
|
|
+48 Transpose perm=(0, 2, 1, 3)
|
|
+24 MatMul
|
|
-12 Attention is_causal=0 kv_num_heads=12 q_num_heads=12 qk_matmul_output_mode=0 softcap=0.0
|
|
+12 Mul
|
|
+12 Softmax axis=-1
|
|
+12 Transpose perm=(0, 1, 3, 2)
|
|
335 nodes, weights 287268b0361f
|
|
DecomposeAttention x156 f5d0980b
|
|
_Fp16TokenEmbedding x2 0d702441
|
|
_ScalarGatherToSlice, _SelectBeforeLayerNorm x1 c46fabc2
|
|
_SelectBeforeLayerNorm x1 eba1fbae
|
|
unstamped x175 ea6c9832
|