alexwengg commited on
Commit
2e263b4
·
verified ·
1 Parent(s): 673ca7d

Upload 16 files

Browse files
cond_step.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e41425950df48be80e3dfd4cff9dc33b0fcd05b1bc97a3dcaa10bbcc86d8c86
3
+ size 243
cond_step.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa2a2c08fd199a50bb5762f9ea0cdd0e5b38673634062c1c955d5b376cb0bfe3
3
+ size 1125
cond_step.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
cond_step.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6378675f49d0e93b9e0536a455c6e43ccd2ca94f63d5e3ebab79af4412f76741
3
+ size 266467520
flow_decoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0149fcce10827c010e495557f287d427f4235a735622edbf09ab5da84c7e1de9
3
+ size 243
flow_decoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc3b49a3f933a737b72e23f1fa9d57d9103a9bffe00ca998011d3ee223c10042
3
+ size 436
flow_decoder.mlmodelc/model.mil ADDED
@@ -0,0 +1,312 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ program(1.3)
2
+ [buildInfo = dict<string, string>({{"coremlc-component-MIL", "3505.3.2"}, {"coremlc-version", "3505.4.1"}, {"coremltools-component-torch", "2.9.1"}, {"coremltools-source-dialect", "TorchScript"}, {"coremltools-version", "9.0"}})]
3
+ {
4
+ func main<ios18>(tensor<fp32, [1, 32]> latent, tensor<fp32, [1, 1]> s, tensor<fp32, [1, 1]> t, tensor<fp32, [1, 1024]> transformer_out) {
5
+ tensor<fp32, [512]> flow_net_input_proj_bias = const()[name = string("flow_net_input_proj_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(64)))];
6
+ tensor<fp32, [512, 32]> flow_net_input_proj_weight = const()[name = string("flow_net_input_proj_weight"), val = tensor<fp32, [512, 32]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2176)))];
7
+ tensor<fp32, [512]> flow_net_time_embed_0_mlp_0_bias = const()[name = string("flow_net_time_embed_0_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(67776)))];
8
+ tensor<fp32, [512, 256]> flow_net_time_embed_0_mlp_0_weight = const()[name = string("flow_net_time_embed_0_mlp_0_weight"), val = tensor<fp32, [512, 256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(69888)))];
9
+ tensor<fp32, [512]> flow_net_time_embed_0_mlp_2_bias = const()[name = string("flow_net_time_embed_0_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(594240)))];
10
+ tensor<fp32, [512, 512]> flow_net_time_embed_0_mlp_2_weight = const()[name = string("flow_net_time_embed_0_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(596352)))];
11
+ tensor<fp32, [512]> flow_net_time_embed_1_mlp_0_bias = const()[name = string("flow_net_time_embed_1_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1644992)))];
12
+ tensor<fp32, [512, 256]> flow_net_time_embed_1_mlp_0_weight = const()[name = string("flow_net_time_embed_1_mlp_0_weight"), val = tensor<fp32, [512, 256]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(1647104)))];
13
+ tensor<fp32, [512]> flow_net_time_embed_1_mlp_2_bias = const()[name = string("flow_net_time_embed_1_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2171456)))];
14
+ tensor<fp32, [512, 512]> flow_net_time_embed_1_mlp_2_weight = const()[name = string("flow_net_time_embed_1_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(2173568)))];
15
+ tensor<fp32, [512]> flow_net_cond_embed_bias = const()[name = string("flow_net_cond_embed_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3222208)))];
16
+ tensor<fp32, [512, 1024]> flow_net_cond_embed_weight = const()[name = string("flow_net_cond_embed_weight"), val = tensor<fp32, [512, 1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(3224320)))];
17
+ tensor<fp32, [1536]> flow_net_res_blocks_0_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_0_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5321536)))];
18
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_0_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_0_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(5327744)))];
19
+ tensor<fp32, [512]> flow_net_res_blocks_0_in_ln_bias = const()[name = string("flow_net_res_blocks_0_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8473536)))];
20
+ tensor<fp32, [512]> flow_net_res_blocks_0_in_ln_weight = const()[name = string("flow_net_res_blocks_0_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8475648)))];
21
+ tensor<fp32, [512]> flow_net_res_blocks_0_mlp_0_bias = const()[name = string("flow_net_res_blocks_0_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8477760)))];
22
+ tensor<fp32, [512, 512]> flow_net_res_blocks_0_mlp_0_weight = const()[name = string("flow_net_res_blocks_0_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(8479872)))];
23
+ tensor<fp32, [512]> flow_net_res_blocks_0_mlp_2_bias = const()[name = string("flow_net_res_blocks_0_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9528512)))];
24
+ tensor<fp32, [512, 512]> flow_net_res_blocks_0_mlp_2_weight = const()[name = string("flow_net_res_blocks_0_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(9530624)))];
25
+ tensor<fp32, [1536]> flow_net_res_blocks_1_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_1_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10579264)))];
26
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_1_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_1_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(10585472)))];
27
+ tensor<fp32, [512]> flow_net_res_blocks_1_in_ln_bias = const()[name = string("flow_net_res_blocks_1_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13731264)))];
28
+ tensor<fp32, [512]> flow_net_res_blocks_1_in_ln_weight = const()[name = string("flow_net_res_blocks_1_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13733376)))];
29
+ tensor<fp32, [512]> flow_net_res_blocks_1_mlp_0_bias = const()[name = string("flow_net_res_blocks_1_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13735488)))];
30
+ tensor<fp32, [512, 512]> flow_net_res_blocks_1_mlp_0_weight = const()[name = string("flow_net_res_blocks_1_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(13737600)))];
31
+ tensor<fp32, [512]> flow_net_res_blocks_1_mlp_2_bias = const()[name = string("flow_net_res_blocks_1_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14786240)))];
32
+ tensor<fp32, [512, 512]> flow_net_res_blocks_1_mlp_2_weight = const()[name = string("flow_net_res_blocks_1_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(14788352)))];
33
+ tensor<fp32, [1536]> flow_net_res_blocks_2_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_2_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15836992)))];
34
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_2_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_2_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(15843200)))];
35
+ tensor<fp32, [512]> flow_net_res_blocks_2_in_ln_bias = const()[name = string("flow_net_res_blocks_2_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(18988992)))];
36
+ tensor<fp32, [512]> flow_net_res_blocks_2_in_ln_weight = const()[name = string("flow_net_res_blocks_2_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(18991104)))];
37
+ tensor<fp32, [512]> flow_net_res_blocks_2_mlp_0_bias = const()[name = string("flow_net_res_blocks_2_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(18993216)))];
38
+ tensor<fp32, [512, 512]> flow_net_res_blocks_2_mlp_0_weight = const()[name = string("flow_net_res_blocks_2_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(18995328)))];
39
+ tensor<fp32, [512]> flow_net_res_blocks_2_mlp_2_bias = const()[name = string("flow_net_res_blocks_2_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(20043968)))];
40
+ tensor<fp32, [512, 512]> flow_net_res_blocks_2_mlp_2_weight = const()[name = string("flow_net_res_blocks_2_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(20046080)))];
41
+ tensor<fp32, [1536]> flow_net_res_blocks_3_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_3_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21094720)))];
42
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_3_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_3_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(21100928)))];
43
+ tensor<fp32, [512]> flow_net_res_blocks_3_in_ln_bias = const()[name = string("flow_net_res_blocks_3_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(24246720)))];
44
+ tensor<fp32, [512]> flow_net_res_blocks_3_in_ln_weight = const()[name = string("flow_net_res_blocks_3_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(24248832)))];
45
+ tensor<fp32, [512]> flow_net_res_blocks_3_mlp_0_bias = const()[name = string("flow_net_res_blocks_3_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(24250944)))];
46
+ tensor<fp32, [512, 512]> flow_net_res_blocks_3_mlp_0_weight = const()[name = string("flow_net_res_blocks_3_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(24253056)))];
47
+ tensor<fp32, [512]> flow_net_res_blocks_3_mlp_2_bias = const()[name = string("flow_net_res_blocks_3_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25301696)))];
48
+ tensor<fp32, [512, 512]> flow_net_res_blocks_3_mlp_2_weight = const()[name = string("flow_net_res_blocks_3_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(25303808)))];
49
+ tensor<fp32, [1536]> flow_net_res_blocks_4_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_4_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(26352448)))];
50
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_4_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_4_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(26358656)))];
51
+ tensor<fp32, [512]> flow_net_res_blocks_4_in_ln_bias = const()[name = string("flow_net_res_blocks_4_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29504448)))];
52
+ tensor<fp32, [512]> flow_net_res_blocks_4_in_ln_weight = const()[name = string("flow_net_res_blocks_4_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29506560)))];
53
+ tensor<fp32, [512]> flow_net_res_blocks_4_mlp_0_bias = const()[name = string("flow_net_res_blocks_4_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29508672)))];
54
+ tensor<fp32, [512, 512]> flow_net_res_blocks_4_mlp_0_weight = const()[name = string("flow_net_res_blocks_4_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(29510784)))];
55
+ tensor<fp32, [512]> flow_net_res_blocks_4_mlp_2_bias = const()[name = string("flow_net_res_blocks_4_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(30559424)))];
56
+ tensor<fp32, [512, 512]> flow_net_res_blocks_4_mlp_2_weight = const()[name = string("flow_net_res_blocks_4_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(30561536)))];
57
+ tensor<fp32, [1536]> flow_net_res_blocks_5_adaLN_modulation_1_bias = const()[name = string("flow_net_res_blocks_5_adaLN_modulation_1_bias"), val = tensor<fp32, [1536]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(31610176)))];
58
+ tensor<fp32, [1536, 512]> flow_net_res_blocks_5_adaLN_modulation_1_weight = const()[name = string("flow_net_res_blocks_5_adaLN_modulation_1_weight"), val = tensor<fp32, [1536, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(31616384)))];
59
+ tensor<fp32, [512]> flow_net_res_blocks_5_in_ln_bias = const()[name = string("flow_net_res_blocks_5_in_ln_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(34762176)))];
60
+ tensor<fp32, [512]> flow_net_res_blocks_5_in_ln_weight = const()[name = string("flow_net_res_blocks_5_in_ln_weight"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(34764288)))];
61
+ tensor<fp32, [512]> flow_net_res_blocks_5_mlp_0_bias = const()[name = string("flow_net_res_blocks_5_mlp_0_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(34766400)))];
62
+ tensor<fp32, [512, 512]> flow_net_res_blocks_5_mlp_0_weight = const()[name = string("flow_net_res_blocks_5_mlp_0_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(34768512)))];
63
+ tensor<fp32, [512]> flow_net_res_blocks_5_mlp_2_bias = const()[name = string("flow_net_res_blocks_5_mlp_2_bias"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(35817152)))];
64
+ tensor<fp32, [512, 512]> flow_net_res_blocks_5_mlp_2_weight = const()[name = string("flow_net_res_blocks_5_mlp_2_weight"), val = tensor<fp32, [512, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(35819264)))];
65
+ tensor<fp32, [1024]> flow_net_final_layer_adaLN_modulation_1_bias = const()[name = string("flow_net_final_layer_adaLN_modulation_1_bias"), val = tensor<fp32, [1024]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(36867904)))];
66
+ tensor<fp32, [1024, 512]> flow_net_final_layer_adaLN_modulation_1_weight = const()[name = string("flow_net_final_layer_adaLN_modulation_1_weight"), val = tensor<fp32, [1024, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(36872064)))];
67
+ tensor<fp32, [32]> flow_net_final_layer_linear_bias = const()[name = string("flow_net_final_layer_linear_bias"), val = tensor<fp32, [32]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38969280)))];
68
+ tensor<fp32, [32, 512]> flow_net_final_layer_linear_weight = const()[name = string("flow_net_final_layer_linear_weight"), val = tensor<fp32, [32, 512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(38969472)))];
69
+ int32 var_9 = const()[name = string("op_9"), val = int32(-1)];
70
+ tensor<fp32, [1, 512]> x_5 = linear(bias = flow_net_input_proj_bias, weight = flow_net_input_proj_weight, x = latent)[name = string("linear_0")];
71
+ tensor<fp32, [128]> const_0 = const()[name = string("const_0"), val = tensor<fp32, [128]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(39035072)))];
72
+ tensor<fp32, [1, 128]> args_1 = mul(x = s, y = const_0)[name = string("args_1")];
73
+ tensor<fp32, [1, 128]> var_39 = cos(x = args_1)[name = string("op_39")];
74
+ tensor<fp32, [1, 128]> var_40 = sin(x = args_1)[name = string("op_40")];
75
+ bool input_1_interleave_0 = const()[name = string("input_1_interleave_0"), val = bool(false)];
76
+ tensor<fp32, [1, 256]> input_1 = concat(axis = var_9, interleave = input_1_interleave_0, values = (var_39, var_40))[name = string("input_1")];
77
+ tensor<fp32, [1, 512]> input_3 = linear(bias = flow_net_time_embed_0_mlp_0_bias, weight = flow_net_time_embed_0_mlp_0_weight, x = input_1)[name = string("linear_1")];
78
+ tensor<fp32, [1, 512]> input_5 = silu(x = input_3)[name = string("input_5")];
79
+ tensor<fp32, [1, 512]> x_1 = linear(bias = flow_net_time_embed_0_mlp_2_bias, weight = flow_net_time_embed_0_mlp_2_weight, x = input_5)[name = string("linear_2")];
80
+ tensor<int32, [1]> reduce_mean_0_axes_0 = const()[name = string("reduce_mean_0_axes_0"), val = tensor<int32, [1]>([-1])];
81
+ bool reduce_mean_0_keep_dims_0 = const()[name = string("reduce_mean_0_keep_dims_0"), val = bool(true)];
82
+ tensor<fp32, [1, 1]> reduce_mean_0 = reduce_mean(axes = reduce_mean_0_axes_0, keep_dims = reduce_mean_0_keep_dims_0, x = x_1)[name = string("reduce_mean_0")];
83
+ tensor<fp32, [1, 512]> sub_0 = sub(x = x_1, y = reduce_mean_0)[name = string("sub_0")];
84
+ tensor<fp32, [1, 512]> square_0 = square(x = sub_0)[name = string("square_0")];
85
+ tensor<int32, [1]> reduce_mean_1_axes_0 = const()[name = string("reduce_mean_1_axes_0"), val = tensor<int32, [1]>([-1])];
86
+ bool reduce_mean_1_keep_dims_0 = const()[name = string("reduce_mean_1_keep_dims_0"), val = bool(true)];
87
+ tensor<fp32, [1, 1]> reduce_mean_1 = reduce_mean(axes = reduce_mean_1_axes_0, keep_dims = reduce_mean_1_keep_dims_0, x = square_0)[name = string("reduce_mean_1")];
88
+ fp32 real_div_0 = const()[name = string("real_div_0"), val = fp32(0x1.00804p+0)];
89
+ tensor<fp32, [1, 1]> mul_0 = mul(x = reduce_mean_1, y = real_div_0)[name = string("mul_0")];
90
+ fp32 var_56 = const()[name = string("op_56"), val = fp32(0x1.4f8b58p-17)];
91
+ tensor<fp32, [1, 1]> var_1 = add(x = mul_0, y = var_56)[name = string("var_1")];
92
+ tensor<fp32, [512]> const_1 = const()[name = string("const_1"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(39035648)))];
93
+ fp32 var_59_epsilon_0 = const()[name = string("op_59_epsilon_0"), val = fp32(0x1.197998p-40)];
94
+ tensor<fp32, [1, 1]> var_59 = rsqrt(epsilon = var_59_epsilon_0, x = var_1)[name = string("op_59")];
95
+ tensor<fp32, [1, 512]> var_60 = mul(x = const_1, y = var_59)[name = string("op_60")];
96
+ tensor<fp32, [1, 512]> var_61 = mul(x = x_1, y = var_60)[name = string("op_61")];
97
+ tensor<fp32, [1, 128]> args = mul(x = t, y = const_0)[name = string("args")];
98
+ tensor<fp32, [1, 128]> var_69 = cos(x = args)[name = string("op_69")];
99
+ tensor<fp32, [1, 128]> var_70 = sin(x = args)[name = string("op_70")];
100
+ bool input_7_interleave_0 = const()[name = string("input_7_interleave_0"), val = bool(false)];
101
+ tensor<fp32, [1, 256]> input_7 = concat(axis = var_9, interleave = input_7_interleave_0, values = (var_69, var_70))[name = string("input_7")];
102
+ tensor<fp32, [1, 512]> input_9 = linear(bias = flow_net_time_embed_1_mlp_0_bias, weight = flow_net_time_embed_1_mlp_0_weight, x = input_7)[name = string("linear_3")];
103
+ tensor<fp32, [1, 512]> input_11 = silu(x = input_9)[name = string("input_11")];
104
+ tensor<fp32, [1, 512]> x_3 = linear(bias = flow_net_time_embed_1_mlp_2_bias, weight = flow_net_time_embed_1_mlp_2_weight, x = input_11)[name = string("linear_4")];
105
+ tensor<int32, [1]> reduce_mean_2_axes_0 = const()[name = string("reduce_mean_2_axes_0"), val = tensor<int32, [1]>([-1])];
106
+ bool reduce_mean_2_keep_dims_0 = const()[name = string("reduce_mean_2_keep_dims_0"), val = bool(true)];
107
+ tensor<fp32, [1, 1]> reduce_mean_2 = reduce_mean(axes = reduce_mean_2_axes_0, keep_dims = reduce_mean_2_keep_dims_0, x = x_3)[name = string("reduce_mean_2")];
108
+ tensor<fp32, [1, 512]> sub_2 = sub(x = x_3, y = reduce_mean_2)[name = string("sub_2")];
109
+ tensor<fp32, [1, 512]> square_1 = square(x = sub_2)[name = string("square_1")];
110
+ tensor<int32, [1]> reduce_mean_3_axes_0 = const()[name = string("reduce_mean_3_axes_0"), val = tensor<int32, [1]>([-1])];
111
+ bool reduce_mean_3_keep_dims_0 = const()[name = string("reduce_mean_3_keep_dims_0"), val = bool(true)];
112
+ tensor<fp32, [1, 1]> reduce_mean_3 = reduce_mean(axes = reduce_mean_3_axes_0, keep_dims = reduce_mean_3_keep_dims_0, x = square_1)[name = string("reduce_mean_3")];
113
+ fp32 real_div_1 = const()[name = string("real_div_1"), val = fp32(0x1.00804p+0)];
114
+ tensor<fp32, [1, 1]> mul_1 = mul(x = reduce_mean_3, y = real_div_1)[name = string("mul_1")];
115
+ fp32 var_86 = const()[name = string("op_86"), val = fp32(0x1.4f8b58p-17)];
116
+ tensor<fp32, [1, 1]> var_3 = add(x = mul_1, y = var_86)[name = string("var_3")];
117
+ tensor<fp32, [512]> const_3 = const()[name = string("const_3"), val = tensor<fp32, [512]>(BLOBFILE(path = string("@model_path/weights/weight.bin"), offset = uint64(39037760)))];
118
+ fp32 var_89_epsilon_0 = const()[name = string("op_89_epsilon_0"), val = fp32(0x1.197998p-40)];
119
+ tensor<fp32, [1, 1]> var_89 = rsqrt(epsilon = var_89_epsilon_0, x = var_3)[name = string("op_89")];
120
+ tensor<fp32, [1, 512]> var_90 = mul(x = const_3, y = var_89)[name = string("op_90")];
121
+ tensor<fp32, [1, 512]> var_91 = mul(x = x_3, y = var_90)[name = string("op_91")];
122
+ tensor<fp32, [1, 512]> var_93 = add(x = var_61, y = var_91)[name = string("op_93")];
123
+ fp32 _inversed_t_combined_y_0 = const()[name = string("_inversed_t_combined_y_0"), val = fp32(0x1p-1)];
124
+ tensor<fp32, [1, 512]> _inversed_t_combined = mul(x = var_93, y = _inversed_t_combined_y_0)[name = string("_inversed_t_combined")];
125
+ tensor<fp32, [1, 512]> c = linear(bias = flow_net_cond_embed_bias, weight = flow_net_cond_embed_weight, x = transformer_out)[name = string("linear_5")];
126
+ tensor<fp32, [1, 512]> input_13 = add(x = _inversed_t_combined, y = c)[name = string("input_13")];
127
+ tensor<fp32, [1, 512]> input_15 = silu(x = input_13)[name = string("input_15")];
128
+ tensor<fp32, [1, 1536]> var_107 = linear(bias = flow_net_res_blocks_0_adaLN_modulation_1_bias, weight = flow_net_res_blocks_0_adaLN_modulation_1_weight, x = input_15)[name = string("linear_6")];
129
+ tensor<int32, [3]> var_108_split_sizes_0 = const()[name = string("op_108_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
130
+ int32 var_108_axis_0 = const()[name = string("op_108_axis_0"), val = int32(-1)];
131
+ tensor<fp32, [1, 512]> var_108_0, tensor<fp32, [1, 512]> var_108_1, tensor<fp32, [1, 512]> var_108_2 = split(axis = var_108_axis_0, split_sizes = var_108_split_sizes_0, x = var_107)[name = string("op_108")];
132
+ tensor<int32, [1]> mean_1_axes_0 = const()[name = string("mean_1_axes_0"), val = tensor<int32, [1]>([-1])];
133
+ bool mean_1_keep_dims_0 = const()[name = string("mean_1_keep_dims_0"), val = bool(true)];
134
+ tensor<fp32, [1, 1]> mean_1 = reduce_mean(axes = mean_1_axes_0, keep_dims = mean_1_keep_dims_0, x = x_5)[name = string("mean_1")];
135
+ tensor<fp32, [1, 512]> sub_4 = sub(x = x_5, y = mean_1)[name = string("sub_4")];
136
+ tensor<fp32, [1, 512]> square_2 = square(x = sub_4)[name = string("square_2")];
137
+ tensor<int32, [1]> reduce_mean_5_axes_0 = const()[name = string("reduce_mean_5_axes_0"), val = tensor<int32, [1]>([-1])];
138
+ bool reduce_mean_5_keep_dims_0 = const()[name = string("reduce_mean_5_keep_dims_0"), val = bool(true)];
139
+ tensor<fp32, [1, 1]> reduce_mean_5 = reduce_mean(axes = reduce_mean_5_axes_0, keep_dims = reduce_mean_5_keep_dims_0, x = square_2)[name = string("reduce_mean_5")];
140
+ fp32 var_118 = const()[name = string("op_118"), val = fp32(0x1.0c6f7ap-20)];
141
+ tensor<fp32, [1, 1]> var_119 = add(x = reduce_mean_5, y = var_118)[name = string("op_119")];
142
+ tensor<fp32, [1, 1]> var_120 = sqrt(x = var_119)[name = string("op_120")];
143
+ tensor<fp32, [1, 512]> x_7 = real_div(x = sub_4, y = var_120)[name = string("x_7")];
144
+ tensor<fp32, [1, 512]> var_122 = mul(x = x_7, y = flow_net_res_blocks_0_in_ln_weight)[name = string("op_122")];
145
+ tensor<fp32, [1, 512]> x_9 = add(x = var_122, y = flow_net_res_blocks_0_in_ln_bias)[name = string("x_9")];
146
+ fp32 var_124_promoted = const()[name = string("op_124_promoted"), val = fp32(0x1p+0)];
147
+ tensor<fp32, [1, 512]> var_125 = add(x = var_108_1, y = var_124_promoted)[name = string("op_125")];
148
+ tensor<fp32, [1, 512]> var_126 = mul(x = x_9, y = var_125)[name = string("op_126")];
149
+ tensor<fp32, [1, 512]> input_17 = add(x = var_126, y = var_108_0)[name = string("input_17")];
150
+ tensor<fp32, [1, 512]> input_19 = linear(bias = flow_net_res_blocks_0_mlp_0_bias, weight = flow_net_res_blocks_0_mlp_0_weight, x = input_17)[name = string("linear_7")];
151
+ tensor<fp32, [1, 512]> input_21 = silu(x = input_19)[name = string("input_21")];
152
+ tensor<fp32, [1, 512]> h_1 = linear(bias = flow_net_res_blocks_0_mlp_2_bias, weight = flow_net_res_blocks_0_mlp_2_weight, x = input_21)[name = string("linear_8")];
153
+ tensor<fp32, [1, 512]> var_137 = mul(x = var_108_2, y = h_1)[name = string("op_137")];
154
+ tensor<fp32, [1, 512]> x_11 = add(x = x_5, y = var_137)[name = string("x_11")];
155
+ tensor<fp32, [1, 1536]> var_146 = linear(bias = flow_net_res_blocks_1_adaLN_modulation_1_bias, weight = flow_net_res_blocks_1_adaLN_modulation_1_weight, x = input_15)[name = string("linear_9")];
156
+ tensor<int32, [3]> var_147_split_sizes_0 = const()[name = string("op_147_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
157
+ int32 var_147_axis_0 = const()[name = string("op_147_axis_0"), val = int32(-1)];
158
+ tensor<fp32, [1, 512]> var_147_0, tensor<fp32, [1, 512]> var_147_1, tensor<fp32, [1, 512]> var_147_2 = split(axis = var_147_axis_0, split_sizes = var_147_split_sizes_0, x = var_146)[name = string("op_147")];
159
+ tensor<int32, [1]> mean_3_axes_0 = const()[name = string("mean_3_axes_0"), val = tensor<int32, [1]>([-1])];
160
+ bool mean_3_keep_dims_0 = const()[name = string("mean_3_keep_dims_0"), val = bool(true)];
161
+ tensor<fp32, [1, 1]> mean_3 = reduce_mean(axes = mean_3_axes_0, keep_dims = mean_3_keep_dims_0, x = x_11)[name = string("mean_3")];
162
+ tensor<fp32, [1, 512]> sub_5 = sub(x = x_11, y = mean_3)[name = string("sub_5")];
163
+ tensor<fp32, [1, 512]> square_3 = square(x = sub_5)[name = string("square_3")];
164
+ tensor<int32, [1]> reduce_mean_7_axes_0 = const()[name = string("reduce_mean_7_axes_0"), val = tensor<int32, [1]>([-1])];
165
+ bool reduce_mean_7_keep_dims_0 = const()[name = string("reduce_mean_7_keep_dims_0"), val = bool(true)];
166
+ tensor<fp32, [1, 1]> reduce_mean_7 = reduce_mean(axes = reduce_mean_7_axes_0, keep_dims = reduce_mean_7_keep_dims_0, x = square_3)[name = string("reduce_mean_7")];
167
+ fp32 var_157 = const()[name = string("op_157"), val = fp32(0x1.0c6f7ap-20)];
168
+ tensor<fp32, [1, 1]> var_158 = add(x = reduce_mean_7, y = var_157)[name = string("op_158")];
169
+ tensor<fp32, [1, 1]> var_159 = sqrt(x = var_158)[name = string("op_159")];
170
+ tensor<fp32, [1, 512]> x_13 = real_div(x = sub_5, y = var_159)[name = string("x_13")];
171
+ tensor<fp32, [1, 512]> var_161 = mul(x = x_13, y = flow_net_res_blocks_1_in_ln_weight)[name = string("op_161")];
172
+ tensor<fp32, [1, 512]> x_15 = add(x = var_161, y = flow_net_res_blocks_1_in_ln_bias)[name = string("x_15")];
173
+ fp32 var_163_promoted = const()[name = string("op_163_promoted"), val = fp32(0x1p+0)];
174
+ tensor<fp32, [1, 512]> var_164 = add(x = var_147_1, y = var_163_promoted)[name = string("op_164")];
175
+ tensor<fp32, [1, 512]> var_165 = mul(x = x_15, y = var_164)[name = string("op_165")];
176
+ tensor<fp32, [1, 512]> input_25 = add(x = var_165, y = var_147_0)[name = string("input_25")];
177
+ tensor<fp32, [1, 512]> input_27 = linear(bias = flow_net_res_blocks_1_mlp_0_bias, weight = flow_net_res_blocks_1_mlp_0_weight, x = input_25)[name = string("linear_10")];
178
+ tensor<fp32, [1, 512]> input_29 = silu(x = input_27)[name = string("input_29")];
179
+ tensor<fp32, [1, 512]> h_3 = linear(bias = flow_net_res_blocks_1_mlp_2_bias, weight = flow_net_res_blocks_1_mlp_2_weight, x = input_29)[name = string("linear_11")];
180
+ tensor<fp32, [1, 512]> var_176 = mul(x = var_147_2, y = h_3)[name = string("op_176")];
181
+ tensor<fp32, [1, 512]> x_17 = add(x = x_11, y = var_176)[name = string("x_17")];
182
+ tensor<fp32, [1, 1536]> var_185 = linear(bias = flow_net_res_blocks_2_adaLN_modulation_1_bias, weight = flow_net_res_blocks_2_adaLN_modulation_1_weight, x = input_15)[name = string("linear_12")];
183
+ tensor<int32, [3]> var_186_split_sizes_0 = const()[name = string("op_186_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
184
+ int32 var_186_axis_0 = const()[name = string("op_186_axis_0"), val = int32(-1)];
185
+ tensor<fp32, [1, 512]> var_186_0, tensor<fp32, [1, 512]> var_186_1, tensor<fp32, [1, 512]> var_186_2 = split(axis = var_186_axis_0, split_sizes = var_186_split_sizes_0, x = var_185)[name = string("op_186")];
186
+ tensor<int32, [1]> mean_5_axes_0 = const()[name = string("mean_5_axes_0"), val = tensor<int32, [1]>([-1])];
187
+ bool mean_5_keep_dims_0 = const()[name = string("mean_5_keep_dims_0"), val = bool(true)];
188
+ tensor<fp32, [1, 1]> mean_5 = reduce_mean(axes = mean_5_axes_0, keep_dims = mean_5_keep_dims_0, x = x_17)[name = string("mean_5")];
189
+ tensor<fp32, [1, 512]> sub_6 = sub(x = x_17, y = mean_5)[name = string("sub_6")];
190
+ tensor<fp32, [1, 512]> square_4 = square(x = sub_6)[name = string("square_4")];
191
+ tensor<int32, [1]> reduce_mean_9_axes_0 = const()[name = string("reduce_mean_9_axes_0"), val = tensor<int32, [1]>([-1])];
192
+ bool reduce_mean_9_keep_dims_0 = const()[name = string("reduce_mean_9_keep_dims_0"), val = bool(true)];
193
+ tensor<fp32, [1, 1]> reduce_mean_9 = reduce_mean(axes = reduce_mean_9_axes_0, keep_dims = reduce_mean_9_keep_dims_0, x = square_4)[name = string("reduce_mean_9")];
194
+ fp32 var_196 = const()[name = string("op_196"), val = fp32(0x1.0c6f7ap-20)];
195
+ tensor<fp32, [1, 1]> var_197 = add(x = reduce_mean_9, y = var_196)[name = string("op_197")];
196
+ tensor<fp32, [1, 1]> var_198 = sqrt(x = var_197)[name = string("op_198")];
197
+ tensor<fp32, [1, 512]> x_19 = real_div(x = sub_6, y = var_198)[name = string("x_19")];
198
+ tensor<fp32, [1, 512]> var_200 = mul(x = x_19, y = flow_net_res_blocks_2_in_ln_weight)[name = string("op_200")];
199
+ tensor<fp32, [1, 512]> x_21 = add(x = var_200, y = flow_net_res_blocks_2_in_ln_bias)[name = string("x_21")];
200
+ fp32 var_202_promoted = const()[name = string("op_202_promoted"), val = fp32(0x1p+0)];
201
+ tensor<fp32, [1, 512]> var_203 = add(x = var_186_1, y = var_202_promoted)[name = string("op_203")];
202
+ tensor<fp32, [1, 512]> var_204 = mul(x = x_21, y = var_203)[name = string("op_204")];
203
+ tensor<fp32, [1, 512]> input_33 = add(x = var_204, y = var_186_0)[name = string("input_33")];
204
+ tensor<fp32, [1, 512]> input_35 = linear(bias = flow_net_res_blocks_2_mlp_0_bias, weight = flow_net_res_blocks_2_mlp_0_weight, x = input_33)[name = string("linear_13")];
205
+ tensor<fp32, [1, 512]> input_37 = silu(x = input_35)[name = string("input_37")];
206
+ tensor<fp32, [1, 512]> h_5 = linear(bias = flow_net_res_blocks_2_mlp_2_bias, weight = flow_net_res_blocks_2_mlp_2_weight, x = input_37)[name = string("linear_14")];
207
+ tensor<fp32, [1, 512]> var_215 = mul(x = var_186_2, y = h_5)[name = string("op_215")];
208
+ tensor<fp32, [1, 512]> x_23 = add(x = x_17, y = var_215)[name = string("x_23")];
209
+ tensor<fp32, [1, 1536]> var_224 = linear(bias = flow_net_res_blocks_3_adaLN_modulation_1_bias, weight = flow_net_res_blocks_3_adaLN_modulation_1_weight, x = input_15)[name = string("linear_15")];
210
+ tensor<int32, [3]> var_225_split_sizes_0 = const()[name = string("op_225_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
211
+ int32 var_225_axis_0 = const()[name = string("op_225_axis_0"), val = int32(-1)];
212
+ tensor<fp32, [1, 512]> var_225_0, tensor<fp32, [1, 512]> var_225_1, tensor<fp32, [1, 512]> var_225_2 = split(axis = var_225_axis_0, split_sizes = var_225_split_sizes_0, x = var_224)[name = string("op_225")];
213
+ tensor<int32, [1]> mean_7_axes_0 = const()[name = string("mean_7_axes_0"), val = tensor<int32, [1]>([-1])];
214
+ bool mean_7_keep_dims_0 = const()[name = string("mean_7_keep_dims_0"), val = bool(true)];
215
+ tensor<fp32, [1, 1]> mean_7 = reduce_mean(axes = mean_7_axes_0, keep_dims = mean_7_keep_dims_0, x = x_23)[name = string("mean_7")];
216
+ tensor<fp32, [1, 512]> sub_7 = sub(x = x_23, y = mean_7)[name = string("sub_7")];
217
+ tensor<fp32, [1, 512]> square_5 = square(x = sub_7)[name = string("square_5")];
218
+ tensor<int32, [1]> reduce_mean_11_axes_0 = const()[name = string("reduce_mean_11_axes_0"), val = tensor<int32, [1]>([-1])];
219
+ bool reduce_mean_11_keep_dims_0 = const()[name = string("reduce_mean_11_keep_dims_0"), val = bool(true)];
220
+ tensor<fp32, [1, 1]> reduce_mean_11 = reduce_mean(axes = reduce_mean_11_axes_0, keep_dims = reduce_mean_11_keep_dims_0, x = square_5)[name = string("reduce_mean_11")];
221
+ fp32 var_235 = const()[name = string("op_235"), val = fp32(0x1.0c6f7ap-20)];
222
+ tensor<fp32, [1, 1]> var_236 = add(x = reduce_mean_11, y = var_235)[name = string("op_236")];
223
+ tensor<fp32, [1, 1]> var_237 = sqrt(x = var_236)[name = string("op_237")];
224
+ tensor<fp32, [1, 512]> x_25 = real_div(x = sub_7, y = var_237)[name = string("x_25")];
225
+ tensor<fp32, [1, 512]> var_239 = mul(x = x_25, y = flow_net_res_blocks_3_in_ln_weight)[name = string("op_239")];
226
+ tensor<fp32, [1, 512]> x_27 = add(x = var_239, y = flow_net_res_blocks_3_in_ln_bias)[name = string("x_27")];
227
+ fp32 var_241_promoted = const()[name = string("op_241_promoted"), val = fp32(0x1p+0)];
228
+ tensor<fp32, [1, 512]> var_242 = add(x = var_225_1, y = var_241_promoted)[name = string("op_242")];
229
+ tensor<fp32, [1, 512]> var_243 = mul(x = x_27, y = var_242)[name = string("op_243")];
230
+ tensor<fp32, [1, 512]> input_41 = add(x = var_243, y = var_225_0)[name = string("input_41")];
231
+ tensor<fp32, [1, 512]> input_43 = linear(bias = flow_net_res_blocks_3_mlp_0_bias, weight = flow_net_res_blocks_3_mlp_0_weight, x = input_41)[name = string("linear_16")];
232
+ tensor<fp32, [1, 512]> input_45 = silu(x = input_43)[name = string("input_45")];
233
+ tensor<fp32, [1, 512]> h_7 = linear(bias = flow_net_res_blocks_3_mlp_2_bias, weight = flow_net_res_blocks_3_mlp_2_weight, x = input_45)[name = string("linear_17")];
234
+ tensor<fp32, [1, 512]> var_254 = mul(x = var_225_2, y = h_7)[name = string("op_254")];
235
+ tensor<fp32, [1, 512]> x_29 = add(x = x_23, y = var_254)[name = string("x_29")];
236
+ tensor<fp32, [1, 1536]> var_263 = linear(bias = flow_net_res_blocks_4_adaLN_modulation_1_bias, weight = flow_net_res_blocks_4_adaLN_modulation_1_weight, x = input_15)[name = string("linear_18")];
237
+ tensor<int32, [3]> var_264_split_sizes_0 = const()[name = string("op_264_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
238
+ int32 var_264_axis_0 = const()[name = string("op_264_axis_0"), val = int32(-1)];
239
+ tensor<fp32, [1, 512]> var_264_0, tensor<fp32, [1, 512]> var_264_1, tensor<fp32, [1, 512]> var_264_2 = split(axis = var_264_axis_0, split_sizes = var_264_split_sizes_0, x = var_263)[name = string("op_264")];
240
+ tensor<int32, [1]> mean_9_axes_0 = const()[name = string("mean_9_axes_0"), val = tensor<int32, [1]>([-1])];
241
+ bool mean_9_keep_dims_0 = const()[name = string("mean_9_keep_dims_0"), val = bool(true)];
242
+ tensor<fp32, [1, 1]> mean_9 = reduce_mean(axes = mean_9_axes_0, keep_dims = mean_9_keep_dims_0, x = x_29)[name = string("mean_9")];
243
+ tensor<fp32, [1, 512]> sub_8 = sub(x = x_29, y = mean_9)[name = string("sub_8")];
244
+ tensor<fp32, [1, 512]> square_6 = square(x = sub_8)[name = string("square_6")];
245
+ tensor<int32, [1]> reduce_mean_13_axes_0 = const()[name = string("reduce_mean_13_axes_0"), val = tensor<int32, [1]>([-1])];
246
+ bool reduce_mean_13_keep_dims_0 = const()[name = string("reduce_mean_13_keep_dims_0"), val = bool(true)];
247
+ tensor<fp32, [1, 1]> reduce_mean_13 = reduce_mean(axes = reduce_mean_13_axes_0, keep_dims = reduce_mean_13_keep_dims_0, x = square_6)[name = string("reduce_mean_13")];
248
+ fp32 var_274 = const()[name = string("op_274"), val = fp32(0x1.0c6f7ap-20)];
249
+ tensor<fp32, [1, 1]> var_275 = add(x = reduce_mean_13, y = var_274)[name = string("op_275")];
250
+ tensor<fp32, [1, 1]> var_276 = sqrt(x = var_275)[name = string("op_276")];
251
+ tensor<fp32, [1, 512]> x_31 = real_div(x = sub_8, y = var_276)[name = string("x_31")];
252
+ tensor<fp32, [1, 512]> var_278 = mul(x = x_31, y = flow_net_res_blocks_4_in_ln_weight)[name = string("op_278")];
253
+ tensor<fp32, [1, 512]> x_33 = add(x = var_278, y = flow_net_res_blocks_4_in_ln_bias)[name = string("x_33")];
254
+ fp32 var_280_promoted = const()[name = string("op_280_promoted"), val = fp32(0x1p+0)];
255
+ tensor<fp32, [1, 512]> var_281 = add(x = var_264_1, y = var_280_promoted)[name = string("op_281")];
256
+ tensor<fp32, [1, 512]> var_282 = mul(x = x_33, y = var_281)[name = string("op_282")];
257
+ tensor<fp32, [1, 512]> input_49 = add(x = var_282, y = var_264_0)[name = string("input_49")];
258
+ tensor<fp32, [1, 512]> input_51 = linear(bias = flow_net_res_blocks_4_mlp_0_bias, weight = flow_net_res_blocks_4_mlp_0_weight, x = input_49)[name = string("linear_19")];
259
+ tensor<fp32, [1, 512]> input_53 = silu(x = input_51)[name = string("input_53")];
260
+ tensor<fp32, [1, 512]> h_9 = linear(bias = flow_net_res_blocks_4_mlp_2_bias, weight = flow_net_res_blocks_4_mlp_2_weight, x = input_53)[name = string("linear_20")];
261
+ tensor<fp32, [1, 512]> var_293 = mul(x = var_264_2, y = h_9)[name = string("op_293")];
262
+ tensor<fp32, [1, 512]> x_35 = add(x = x_29, y = var_293)[name = string("x_35")];
263
+ tensor<fp32, [1, 1536]> var_302 = linear(bias = flow_net_res_blocks_5_adaLN_modulation_1_bias, weight = flow_net_res_blocks_5_adaLN_modulation_1_weight, x = input_15)[name = string("linear_21")];
264
+ tensor<int32, [3]> var_303_split_sizes_0 = const()[name = string("op_303_split_sizes_0"), val = tensor<int32, [3]>([512, 512, 512])];
265
+ int32 var_303_axis_0 = const()[name = string("op_303_axis_0"), val = int32(-1)];
266
+ tensor<fp32, [1, 512]> var_303_0, tensor<fp32, [1, 512]> var_303_1, tensor<fp32, [1, 512]> var_303_2 = split(axis = var_303_axis_0, split_sizes = var_303_split_sizes_0, x = var_302)[name = string("op_303")];
267
+ tensor<int32, [1]> mean_11_axes_0 = const()[name = string("mean_11_axes_0"), val = tensor<int32, [1]>([-1])];
268
+ bool mean_11_keep_dims_0 = const()[name = string("mean_11_keep_dims_0"), val = bool(true)];
269
+ tensor<fp32, [1, 1]> mean_11 = reduce_mean(axes = mean_11_axes_0, keep_dims = mean_11_keep_dims_0, x = x_35)[name = string("mean_11")];
270
+ tensor<fp32, [1, 512]> sub_9 = sub(x = x_35, y = mean_11)[name = string("sub_9")];
271
+ tensor<fp32, [1, 512]> square_7 = square(x = sub_9)[name = string("square_7")];
272
+ tensor<int32, [1]> reduce_mean_15_axes_0 = const()[name = string("reduce_mean_15_axes_0"), val = tensor<int32, [1]>([-1])];
273
+ bool reduce_mean_15_keep_dims_0 = const()[name = string("reduce_mean_15_keep_dims_0"), val = bool(true)];
274
+ tensor<fp32, [1, 1]> reduce_mean_15 = reduce_mean(axes = reduce_mean_15_axes_0, keep_dims = reduce_mean_15_keep_dims_0, x = square_7)[name = string("reduce_mean_15")];
275
+ fp32 var_313 = const()[name = string("op_313"), val = fp32(0x1.0c6f7ap-20)];
276
+ tensor<fp32, [1, 1]> var_314 = add(x = reduce_mean_15, y = var_313)[name = string("op_314")];
277
+ tensor<fp32, [1, 1]> var_315 = sqrt(x = var_314)[name = string("op_315")];
278
+ tensor<fp32, [1, 512]> x_37 = real_div(x = sub_9, y = var_315)[name = string("x_37")];
279
+ tensor<fp32, [1, 512]> var_317 = mul(x = x_37, y = flow_net_res_blocks_5_in_ln_weight)[name = string("op_317")];
280
+ tensor<fp32, [1, 512]> x_39 = add(x = var_317, y = flow_net_res_blocks_5_in_ln_bias)[name = string("x_39")];
281
+ fp32 var_319_promoted = const()[name = string("op_319_promoted"), val = fp32(0x1p+0)];
282
+ tensor<fp32, [1, 512]> var_320 = add(x = var_303_1, y = var_319_promoted)[name = string("op_320")];
283
+ tensor<fp32, [1, 512]> var_321 = mul(x = x_39, y = var_320)[name = string("op_321")];
284
+ tensor<fp32, [1, 512]> input_57 = add(x = var_321, y = var_303_0)[name = string("input_57")];
285
+ tensor<fp32, [1, 512]> input_59 = linear(bias = flow_net_res_blocks_5_mlp_0_bias, weight = flow_net_res_blocks_5_mlp_0_weight, x = input_57)[name = string("linear_22")];
286
+ tensor<fp32, [1, 512]> input_61 = silu(x = input_59)[name = string("input_61")];
287
+ tensor<fp32, [1, 512]> h = linear(bias = flow_net_res_blocks_5_mlp_2_bias, weight = flow_net_res_blocks_5_mlp_2_weight, x = input_61)[name = string("linear_23")];
288
+ tensor<fp32, [1, 512]> var_332 = mul(x = var_303_2, y = h)[name = string("op_332")];
289
+ tensor<fp32, [1, 512]> x_41 = add(x = x_35, y = var_332)[name = string("x_41")];
290
+ tensor<fp32, [1, 1024]> var_340 = linear(bias = flow_net_final_layer_adaLN_modulation_1_bias, weight = flow_net_final_layer_adaLN_modulation_1_weight, x = input_15)[name = string("linear_24")];
291
+ tensor<int32, [2]> var_341_split_sizes_0 = const()[name = string("op_341_split_sizes_0"), val = tensor<int32, [2]>([512, 512])];
292
+ int32 var_341_axis_0 = const()[name = string("op_341_axis_0"), val = int32(-1)];
293
+ tensor<fp32, [1, 512]> var_341_0, tensor<fp32, [1, 512]> var_341_1 = split(axis = var_341_axis_0, split_sizes = var_341_split_sizes_0, x = var_340)[name = string("op_341")];
294
+ tensor<int32, [1]> mean_axes_0 = const()[name = string("mean_axes_0"), val = tensor<int32, [1]>([-1])];
295
+ bool mean_keep_dims_0 = const()[name = string("mean_keep_dims_0"), val = bool(true)];
296
+ tensor<fp32, [1, 1]> mean = reduce_mean(axes = mean_axes_0, keep_dims = mean_keep_dims_0, x = x_41)[name = string("mean")];
297
+ tensor<fp32, [1, 512]> sub_10 = sub(x = x_41, y = mean)[name = string("sub_10")];
298
+ tensor<fp32, [1, 512]> square_8 = square(x = sub_10)[name = string("square_8")];
299
+ tensor<int32, [1]> reduce_mean_17_axes_0 = const()[name = string("reduce_mean_17_axes_0"), val = tensor<int32, [1]>([-1])];
300
+ bool reduce_mean_17_keep_dims_0 = const()[name = string("reduce_mean_17_keep_dims_0"), val = bool(true)];
301
+ tensor<fp32, [1, 1]> reduce_mean_17 = reduce_mean(axes = reduce_mean_17_axes_0, keep_dims = reduce_mean_17_keep_dims_0, x = square_8)[name = string("reduce_mean_17")];
302
+ fp32 var_348 = const()[name = string("op_348"), val = fp32(0x1.0c6f7ap-20)];
303
+ tensor<fp32, [1, 1]> var_349 = add(x = reduce_mean_17, y = var_348)[name = string("op_349")];
304
+ tensor<fp32, [1, 1]> var_350 = sqrt(x = var_349)[name = string("op_350")];
305
+ tensor<fp32, [1, 512]> x = real_div(x = sub_10, y = var_350)[name = string("x")];
306
+ fp32 var_352_promoted = const()[name = string("op_352_promoted"), val = fp32(0x1p+0)];
307
+ tensor<fp32, [1, 512]> var_353 = add(x = var_341_1, y = var_352_promoted)[name = string("op_353")];
308
+ tensor<fp32, [1, 512]> var_354 = mul(x = x, y = var_353)[name = string("op_354")];
309
+ tensor<fp32, [1, 512]> input = add(x = var_354, y = var_341_0)[name = string("input")];
310
+ tensor<fp32, [1, 32]> var_358 = linear(bias = flow_net_final_layer_linear_bias, weight = flow_net_final_layer_linear_weight, x = input)[name = string("linear_25")];
311
+ } -> (var_358);
312
+ }
flow_decoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fa32f175b149cbf002d19eb8f72c4421b091ca8bead3c5d2687696966b5d9ca
3
+ size 39039872
flowlm_step.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa1ec42ed08d675b9c5b29875604cfc16e9a0a47b3ad84809680ecfefc35e524
3
+ size 243
flowlm_step.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6f29215ac1a668598e3a98254836cc2f12710f9e39a0a2f782b91128f31b8f9
3
+ size 1190
flowlm_step.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
flowlm_step.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e52a33521c2e98d1399729636733c7371a8f29c365ac66162b9eeaa43c3d1ff
3
+ size 304368384
mimi_decoder.mlmodelc/analytics/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03bcfd00f98f70216031b359dfd3459d8ddcbfb223ac0fb0297f9c76bef1fb05
3
+ size 243
mimi_decoder.mlmodelc/coremldata.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4306ebfc373705e4e7f9e0d1e7f9e5c992134f5e5b4a7625b43421af5b3956c6
3
+ size 1714
mimi_decoder.mlmodelc/model.mil ADDED
The diff for this file is too large to render. See raw diff
 
mimi_decoder.mlmodelc/weights/weight.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d184d69a4ffe9dc47084da8297916a8d6a78889fee7878f1b964373d7794c05f
3
+ size 20591680