supra-tex / completion /config.json
supra3d's picture
Upload SUPRA-Tex weights and pipeline manifest
499db28 verified
Raw History Blame Contribute Delete
7.4 kB
{
"format": "supratex_completion",
"schema_version": 1,
"models": {
"denoiser": {
"name": "InterleavedFlowTransformer",
"args": {
"resolution": 64,
"in_channels": 64,
"out_channels": 32,
"model_channels": 1536,
"cond_channels": 0,
"num_blocks": 60,
"v11_init_policy": "odd_copy_even_identity",
"base_block_policy": "alternating_u_x",
"num_heads": 12,
"mlp_ratio": 5.3334,
"rope_freq": [
1,
10000
],
"dtype": "float32",
"use_checkpoint": true,
"checkpoint_group_size": 2,
"share_mod": true,
"qk_rms_norm": true,
"partial_mean": [
3.611770664232335,
1.9163118034138746,
1.952608078723815,
0.2910785085662031,
0.050816326157399436,
-0.5254554342843234,
0.30951655881595963,
-0.8197975459866622,
0.38620928427331275,
-0.4964843134411725,
-0.6825266282042826,
1.3085809901770953,
-1.0911941103735894,
0.44349868701875067,
-1.1721900675779473,
1.1157969135230572,
-1.135226667830614,
2.234862037712815,
0.28582271446763646,
2.640323985270555,
0.15107635964282012,
2.165778925496729,
-1.7239707588343525,
-1.4536538929673588,
0.9615435201854913,
0.9627418033316325,
0.17127697540739928,
-0.044568065854129285,
2.4527678131314383,
0.23741388043728626,
-0.7125081495349594,
-0.8392259054313957
],
"partial_std": [
2.281878097994697,
2.2069980905383275,
2.301781262493739,
2.2781209635645494,
2.8149015793750904,
1.803326133427939,
1.7410887436600158,
2.523892276683066,
2.7172925946489106,
2.1446440936869227,
1.7059470326455453,
2.70657116251002,
2.386575929607397,
1.9716654166648369,
2.8987434709120956,
2.412686860777434,
1.9001049086001556,
2.8054303614532863,
2.938243007849802,
1.853011888157837,
2.35922996737155,
2.4103167387725697,
3.0474224154374827,
2.386036377355109,
2.2699966056199004,
2.060129709179025,
2.127143704781528,
1.988618505693798,
2.562611683572114,
1.87036480863336,
2.5588610248943686,
2.4600089118392092
],
"pbr_mean": [
3.501659,
2.212398,
2.226094,
0.251093,
-0.026248,
-0.687364,
0.439898,
-0.928075,
0.029398,
-0.339596,
-0.869527,
1.038479,
-0.972385,
0.126042,
-1.129303,
0.455149,
-1.209521,
2.069067,
0.544735,
2.569128,
-0.323407,
2.293,
-1.925608,
-1.217717,
1.213905,
0.971588,
-0.023631,
0.10675,
2.021786,
0.250524,
-0.662387,
-0.768862
],
"pbr_std": [
2.665652,
2.743913,
2.765121,
2.595319,
3.037293,
2.291316,
2.144656,
2.911822,
2.969419,
2.501689,
2.154811,
3.163343,
2.621215,
2.381943,
3.186697,
3.021588,
2.295916,
3.234985,
3.233086,
2.26014,
2.874801,
2.810596,
3.29272,
2.674999,
2.680878,
2.372054,
2.451546,
2.353556,
2.995195,
2.379849,
2.786195,
2.77519
],
"direct_modulation_broadcast": true,
"safe_empty_cache": true,
"refinement_indices": []
}
}
},
"dataset": {
"args": {
"resolution": 1024,
"attrs": [
"base_color",
"metallic",
"roughness",
"alpha"
],
"shape_slat_normalization": {
"mean": [
0.781296,
0.018091,
-0.495192,
-0.558457,
1.06053,
0.093252,
1.518149,
-0.933218,
-0.732996,
2.604095,
-0.118341,
-2.143904,
0.495076,
-2.179512,
-2.130751,
-0.996944,
0.261421,
-2.217463,
1.260067,
-0.150213,
3.790713,
1.481266,
-1.046058,
-1.523667,
-0.059621,
2.22078,
1.621212,
0.87723,
0.567247,
-3.175944,
-3.186688,
1.578665
],
"std": [
5.972266,
4.706852,
5.44501,
5.209927,
5.32022,
4.547237,
5.020802,
5.444004,
5.226681,
5.683095,
4.831436,
5.286469,
5.652043,
5.367606,
5.525084,
4.730578,
4.805265,
5.124013,
5.530808,
5.619001,
5.10393,
5.41767,
5.269677,
5.547194,
5.634698,
5.235274,
6.110351,
5.511298,
6.237273,
4.879207,
5.347008,
5.405691
]
},
"pbr_slat_normalization": {
"mean": [
3.501659,
2.212398,
2.226094,
0.251093,
-0.026248,
-0.687364,
0.439898,
-0.928075,
0.029398,
-0.339596,
-0.869527,
1.038479,
-0.972385,
0.126042,
-1.129303,
0.455149,
-1.209521,
2.069067,
0.544735,
2.569128,
-0.323407,
2.293,
-1.925608,
-1.217717,
1.213905,
0.971588,
-0.023631,
0.10675,
2.021786,
0.250524,
-0.662387,
-0.768862
],
"std": [
2.665652,
2.743913,
2.765121,
2.595319,
3.037293,
2.291316,
2.144656,
2.911822,
2.969419,
2.501689,
2.154811,
3.163343,
2.621215,
2.381943,
3.186697,
3.021588,
2.295916,
3.234985,
3.233086,
2.26014,
2.874801,
2.810596,
3.29272,
2.674999,
2.680878,
2.372054,
2.451546,
2.353556,
2.995195,
2.379849,
2.786195,
2.77519
]
},
"bridge_z_obs_args": {
"partial_support_mode": "partial_messages",
"partial_z_obs_normalization_path": "partial_normalization.json"
},
"pretrained_shape_slat_dec": "shape_dec_next_dc_f16c32_fp16",
"pretrained_pbr_slat_dec": "tex_dec_next_dc_f16c32_fp16"
}
}
}