Hugging Face's logo

luweigen
/
moebius-onnx-android

ONNX
inpainting
android
qnn
Model card Files
xet
 Community
moebius-onnx-android / hyb.json
luweigen's picture
luweigen
Upload models for the Android app
4cefe13 verified
Raw History Blame Contribute Delete
6.12 kB
{
"segments": [
{
"file": "hyb.s0.htp.onnx",
"device": "htp",
"nodes": [
0,
81
],
"inputs": [
"context",
"latent"
],
"outputs": [
"/m/diff_model/encoder_hid_proj/Add_output_0",
"/m/diff_model/up_blocks.2/attentions.2/transformer_blocks.0/attn2/Reshape_4_output_0",
"/m/diff_model/up_blocks.2/attentions.2/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/up_blocks.2/attentions.1/transformer_blocks.0/attn2/Reshape_4_output_0",
"/m/diff_model/up_blocks.2/attentions.1/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/up_blocks.2/attentions.0/transformer_blocks.0/attn2/Reshape_4_output_0",
"/m/diff_model/up_blocks.2/attentions.0/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/down_blocks.0/attentions.1/transformer_blocks.0/attn2/Reshape_4_output_0",
"/m/diff_model/down_blocks.0/attentions.1/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/down_blocks.0/attentions.0/transformer_blocks.0/attn2/Reshape_4_output_0",
"/m/diff_model/down_blocks.0/attentions.0/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/conv_in/conv_pw/Conv_output_0",
"/m/diff_model/down_blocks.0/resnets.0/conv1/Add_output_0"
]
},
{
"file": "hyb.s1.cpu.onnx",
"device": "cpu",
"nodes": [
82,
95
],
"inputs": [
"timesteps"
],
"outputs": [
"/m/diff_model/time_embedding/linear_2/Gemm_output_0"
]
},
{
"file": "hyb.s2.htp.onnx",
"device": "htp",
"nodes": [
96,
114
],
"inputs": [
"/m/diff_model/time_embedding/linear_2/Gemm_output_0",
"/m/diff_model/down_blocks.0/resnets.0/conv1/Add_output_0",
"/m/diff_model/conv_in/conv_pw/Conv_output_0"
],
"outputs": [
"/m/diff_model/down_blocks.0/resnets.0/nonlinearity_1/Mul_output_0",
"/m/diff_model/down_blocks.0/resnets.0/Add_1_output_0"
]
},
{
"file": "hyb.s3.cpu.onnx",
"device": "cpu",
"nodes": [
115,
198
],
"inputs": [
"/m/diff_model/down_blocks.0/resnets.0/Add_1_output_0",
"/m/diff_model/down_blocks.0/attentions.0/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/down_blocks.0/attentions.0/transformer_blocks.0/attn2/Reshape_4_output_0"
],
"outputs": [
"/m/diff_model/down_blocks.0/attentions.0/Add_output_0"
]
},
{
"file": "hyb.s4.htp.onnx",
"device": "htp",
"nodes": [
199,
227
],
"inputs": [
"/m/diff_model/down_blocks.0/attentions.0/Add_output_0",
"/m/diff_model/down_blocks.0/resnets.0/nonlinearity_1/Mul_output_0"
],
"outputs": [
"/m/diff_model/down_blocks.0/resnets.1/Add_1_output_0"
]
},
{
"file": "hyb.s5.cpu.onnx",
"device": "cpu",
"nodes": [
228,
310
],
"inputs": [
"/m/diff_model/down_blocks.0/resnets.1/Add_1_output_0",
"/m/diff_model/down_blocks.0/attentions.1/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/down_blocks.0/attentions.1/transformer_blocks.0/attn2/Reshape_4_output_0"
],
"outputs": [
"/m/diff_model/down_blocks.0/attentions.1/Add_output_0"
]
},
{
"file": "hyb.s6.htp.onnx",
"device": "htp",
"nodes": [
311,
1631
],
"inputs": [
"/m/diff_model/down_blocks.0/attentions.1/Add_output_0",
"/m/diff_model/down_blocks.0/resnets.0/nonlinearity_1/Mul_output_0",
"/m/diff_model/encoder_hid_proj/Add_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/resnets.0/Add_1_output_0"
]
},
{
"file": "hyb.s7.cpu.onnx",
"device": "cpu",
"nodes": [
1632,
1714
],
"inputs": [
"/m/diff_model/up_blocks.2/resnets.0/Add_1_output_0",
"/m/diff_model/up_blocks.2/attentions.0/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/up_blocks.2/attentions.0/transformer_blocks.0/attn2/Reshape_4_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/attentions.0/Add_output_0"
]
},
{
"file": "hyb.s8.htp.onnx",
"device": "htp",
"nodes": [
1715,
1744
],
"inputs": [
"/m/diff_model/up_blocks.2/attentions.0/Add_output_0",
"/m/diff_model/down_blocks.0/attentions.0/Add_output_0",
"/m/diff_model/down_blocks.0/resnets.0/nonlinearity_1/Mul_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/resnets.1/Add_1_output_0"
]
},
{
"file": "hyb.s9.cpu.onnx",
"device": "cpu",
"nodes": [
1745,
1827
],
"inputs": [
"/m/diff_model/up_blocks.2/resnets.1/Add_1_output_0",
"/m/diff_model/up_blocks.2/attentions.1/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/up_blocks.2/attentions.1/transformer_blocks.0/attn2/Reshape_4_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/attentions.1/Add_output_0"
]
},
{
"file": "hyb.s10.htp.onnx",
"device": "htp",
"nodes": [
1828,
1857
],
"inputs": [
"/m/diff_model/up_blocks.2/attentions.1/Add_output_0",
"/m/diff_model/conv_in/conv_pw/Conv_output_0",
"/m/diff_model/down_blocks.0/resnets.0/nonlinearity_1/Mul_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/resnets.2/Add_1_output_0"
]
},
{
"file": "hyb.s11.cpu.onnx",
"device": "cpu",
"nodes": [
1858,
1942
],
"inputs": [
"/m/diff_model/up_blocks.2/resnets.2/Add_1_output_0",
"/m/diff_model/up_blocks.2/attentions.2/transformer_blocks.0/attn2/Einsum_1/in2_R",
"/m/diff_model/up_blocks.2/attentions.2/transformer_blocks.0/attn2/Reshape_4_output_0"
],
"outputs": [
"/m/diff_model/up_blocks.2/attentions.2/Add_output_0"
]
},
{
"file": "hyb.s12.htp.onnx",
"device": "htp",
"nodes": [
1943,
1953
],
"inputs": [
"/m/diff_model/up_blocks.2/attentions.2/Add_output_0"
],
"outputs": [
"noise"
]
}
],
"final": "noise",
"graph_inputs": [
"latent",
"timesteps",
"context"
],
"source": "unet.ns.onnx"
}