| | |
| | def get_rgb_factors(model_family, model_type = None, sub_family = None): |
| | if model_family in ["wan", "qwen"]: |
| | if model_type =="ti2v_2_2": |
| | latent_channels = 48 |
| | latent_dimensions = 3 |
| | latent_rgb_factors = [ |
| | [ 0.0119, 0.0103, 0.0046], |
| | [-0.1062, -0.0504, 0.0165], |
| | [ 0.0140, 0.0409, 0.0491], |
| | [-0.0813, -0.0677, 0.0607], |
| | [ 0.0656, 0.0851, 0.0808], |
| | [ 0.0264, 0.0463, 0.0912], |
| | [ 0.0295, 0.0326, 0.0590], |
| | [-0.0244, -0.0270, 0.0025], |
| | [ 0.0443, -0.0102, 0.0288], |
| | [-0.0465, -0.0090, -0.0205], |
| | [ 0.0359, 0.0236, 0.0082], |
| | [-0.0776, 0.0854, 0.1048], |
| | [ 0.0564, 0.0264, 0.0561], |
| | [ 0.0006, 0.0594, 0.0418], |
| | [-0.0319, -0.0542, -0.0637], |
| | [-0.0268, 0.0024, 0.0260], |
| | [ 0.0539, 0.0265, 0.0358], |
| | [-0.0359, -0.0312, -0.0287], |
| | [-0.0285, -0.1032, -0.1237], |
| | [ 0.1041, 0.0537, 0.0622], |
| | [-0.0086, -0.0374, -0.0051], |
| | [ 0.0390, 0.0670, 0.2863], |
| | [ 0.0069, 0.0144, 0.0082], |
| | [ 0.0006, -0.0167, 0.0079], |
| | [ 0.0313, -0.0574, -0.0232], |
| | [-0.1454, -0.0902, -0.0481], |
| | [ 0.0714, 0.0827, 0.0447], |
| | [-0.0304, -0.0574, -0.0196], |
| | [ 0.0401, 0.0384, 0.0204], |
| | [-0.0758, -0.0297, -0.0014], |
| | [ 0.0568, 0.1307, 0.1372], |
| | [-0.0055, -0.0310, -0.0380], |
| | [ 0.0239, -0.0305, 0.0325], |
| | [-0.0663, -0.0673, -0.0140], |
| | [-0.0416, -0.0047, -0.0023], |
| | [ 0.0166, 0.0112, -0.0093], |
| | [-0.0211, 0.0011, 0.0331], |
| | [ 0.1833, 0.1466, 0.2250], |
| | [-0.0368, 0.0370, 0.0295], |
| | [-0.3441, -0.3543, -0.2008], |
| | [-0.0479, -0.0489, -0.0420], |
| | [-0.0660, -0.0153, 0.0800], |
| | [-0.0101, 0.0068, 0.0156], |
| | [-0.0690, -0.0452, -0.0927], |
| | [-0.0145, 0.0041, 0.0015], |
| | [ 0.0421, 0.0451, 0.0373], |
| | [ 0.0504, -0.0483, -0.0356], |
| | [-0.0837, 0.0168, 0.0055] |
| | ] |
| | else: |
| | latent_channels = 16 |
| | latent_dimensions = 3 |
| | latent_rgb_factors = [ |
| | [-0.1299, -0.1692, 0.2932], |
| | [ 0.0671, 0.0406, 0.0442], |
| | [ 0.3568, 0.2548, 0.1747], |
| | [ 0.0372, 0.2344, 0.1420], |
| | [ 0.0313, 0.0189, -0.0328], |
| | [ 0.0296, -0.0956, -0.0665], |
| | [-0.3477, -0.4059, -0.2925], |
| | [ 0.0166, 0.1902, 0.1975], |
| | [-0.0412, 0.0267, -0.1364], |
| | [-0.1293, 0.0740, 0.1636], |
| | [ 0.0680, 0.3019, 0.1128], |
| | [ 0.0032, 0.0581, 0.0639], |
| | [-0.1251, 0.0927, 0.1699], |
| | [ 0.0060, -0.0633, 0.0005], |
| | [ 0.3477, 0.2275, 0.2950], |
| | [ 0.1984, 0.0913, 0.1861] |
| | ] |
| | |
| | latent_rgb_factors_bias = [-0.1835, -0.0868, -0.3360] |
| |
|
| | |
| | elif model_family =="flux": |
| | if sub_family =="flux2": |
| | |
| | latent_rgb_factors = [ |
| | [0.017027, 0.008257, 0.004624], |
| | [0.026682, 0.029259, 0.060114], |
| | [-0.009869, 0.020518, 0.060890], |
| | [0.226466, 0.296887, 0.368866], |
| | [0.015064, -0.014507, -0.028432], |
| | [0.001710, -0.004815, -0.022213], |
| | [-0.018906, 0.049969, -0.008771], |
| | [0.010261, 0.012761, 0.017796], |
| | [-0.349602, -0.272936, -0.182633], |
| | [-0.031123, -0.042880, -0.043945], |
| | [-0.008367, -0.001142, -0.001056], |
| | [0.036070, -0.005392, -0.069563], |
| | [-0.003169, 0.013091, 0.029921], |
| | [-0.008074, -0.009231, -0.010962], |
| | [0.008431, 0.015439, 0.015267], |
| | [-0.066895, -0.005198, 0.011012], |
| | [0.012917, 0.009686, 0.012514], |
| | [-0.001605, -0.000442, 0.002637], |
| | [0.012599, 0.011981, 0.014867], |
| | [0.068052, -0.003048, -0.001778], |
| | [-0.002239, 0.003910, 0.009552], |
| | [0.010798, 0.000513, -0.002564], |
| | [0.013768, 0.016439, 0.016339], |
| | [-0.004254, 0.003370, 0.012017], |
| | [-0.048989, -0.029049, -0.032899], |
| | [-0.009327, -0.010332, -0.004374], |
| | [0.007945, -0.008562, -0.006007], |
| | [-0.002106, -0.018114, -0.029735], |
| | [-0.039313, -0.046827, -0.048615], |
| | [-0.005774, 0.016549, 0.031351], |
| | [0.005425, 0.001212, -0.003960], |
| | [-0.009495, -0.028664, -0.039440], |
| | ] |
| | latent_rgb_factors_bias = [-0.039388, -0.063196, -0.118529] |
| | else: |
| | scale_factor = 0.3611 |
| | shift_factor = 0.1159 |
| | latent_rgb_factors =[ |
| | [-0.0346, 0.0244, 0.0681], |
| | [ 0.0034, 0.0210, 0.0687], |
| | [ 0.0275, -0.0668, -0.0433], |
| | [-0.0174, 0.0160, 0.0617], |
| | [ 0.0859, 0.0721, 0.0329], |
| | [ 0.0004, 0.0383, 0.0115], |
| | [ 0.0405, 0.0861, 0.0915], |
| | [-0.0236, -0.0185, -0.0259], |
| | [-0.0245, 0.0250, 0.1180], |
| | [ 0.1008, 0.0755, -0.0421], |
| | [-0.0515, 0.0201, 0.0011], |
| | [ 0.0428, -0.0012, -0.0036], |
| | [ 0.0817, 0.0765, 0.0749], |
| | [-0.1264, -0.0522, -0.1103], |
| | [-0.0280, -0.0881, -0.0499], |
| | [-0.1262, -0.0982, -0.0778] |
| | ] |
| | latent_rgb_factors_bias = [-0.0329, -0.0718, -0.0851] |
| |
|
| | elif model_family == "ltx2": |
| | |
| | latent_rgb_factors = [ |
| | [0.034942, 0.025665, 0.019890], |
| | [-0.010039, -0.003935, -0.004114], |
| | [0.012398, -0.009390, 0.002166], |
| | [0.021493, 0.004525, -0.001346], |
| | [0.007121, 0.021149, 0.006106], |
| | [0.009438, 0.013676, 0.003909], |
| | [-0.012689, -0.017296, -0.011732], |
| | [-0.010145, -0.008827, -0.040893], |
| | [-0.000609, 0.002179, 0.004511], |
| | [0.017091, 0.010640, 0.000340], |
| | [-0.009421, -0.006105, 0.000176], |
| | [-0.005993, -0.002687, 0.003946], |
| | [0.014557, 0.021932, 0.020061], |
| | [0.008345, 0.001006, -0.004580], |
| | [0.034493, 0.018170, 0.023638], |
| | [-0.001190, 0.003468, 0.012947], |
| | [-0.002343, -0.012411, -0.011100], |
| | [-0.011259, -0.000162, -0.005704], |
| | [0.035304, 0.037420, 0.045084], |
| | [-0.002184, -0.026547, -0.019204], |
| | [0.011548, 0.006255, 0.012846], |
| | [-0.000233, -0.002465, -0.004693], |
| | [-0.034094, -0.021138, -0.005063], |
| | [-0.000989, -0.000375, 0.002170], |
| | [0.007333, 0.015489, 0.024374], |
| | [0.006371, -0.018043, -0.031470], |
| | [-0.029347, -0.027023, -0.016491], |
| | [-0.001422, -0.006893, -0.001695], |
| | [0.005795, 0.005987, 0.007621], |
| | [-0.003406, 0.002260, 0.000005], |
| | [0.029885, 0.015924, 0.013307], |
| | [0.010122, 0.005565, 0.003611], |
| | [0.010770, 0.001935, 0.003951], |
| | [0.000247, 0.007556, -0.000721], |
| | [0.003422, 0.002761, 0.002203], |
| | [-0.000763, -0.001199, 0.002576], |
| | [0.013336, 0.018443, 0.022138], |
| | [-0.011958, -0.026991, -0.033736], |
| | [0.000398, -0.005840, -0.001664], |
| | [0.001106, 0.001808, -0.001595], |
| | [-0.003785, 0.003479, -0.001016], |
| | [0.015053, 0.015471, 0.013186], |
| | [0.008358, 0.007849, 0.018877], |
| | [0.005239, -0.003565, -0.034033], |
| | [-0.055078, -0.052277, -0.043015], |
| | [-0.004671, -0.017110, -0.012464], |
| | [-0.007845, -0.001926, -0.000054], |
| | [0.063222, 0.078728, 0.058501], |
| | [-0.021986, -0.016549, -0.020792], |
| | [0.000084, -0.003974, -0.002621], |
| | [-0.008120, -0.009763, -0.020849], |
| | [0.008795, 0.008390, 0.004435], |
| | [0.003783, -0.002978, -0.003743], |
| | [0.013151, -0.004677, -0.001796], |
| | [0.002784, 0.000578, 0.012341], |
| | [0.032561, 0.018570, 0.015776], |
| | [-0.018336, -0.014570, 0.001422], |
| | [-0.016377, -0.004534, -0.008379], |
| | [-0.018032, -0.018326, -0.016427], |
| | [0.011842, 0.012533, 0.018174], |
| | [-0.007721, -0.007604, -0.009647], |
| | [0.010696, 0.002887, 0.012046], |
| | [-0.022643, -0.029499, -0.022168], |
| | [-0.004399, -0.003200, -0.001333], |
| | [0.077523, 0.061106, 0.040685], |
| | [0.016623, 0.018785, 0.015607], |
| | [0.005365, 0.004560, -0.008725], |
| | [0.001207, -0.001362, 0.001578], |
| | [0.019177, 0.030301, 0.023738], |
| | [0.000936, 0.001365, 0.005611], |
| | [0.015523, 0.008943, 0.012178], |
| | [-0.026548, -0.002319, 0.008205], |
| | [0.015405, -0.008554, -0.012774], |
| | [-0.002702, 0.012692, 0.004927], |
| | [-0.001287, -0.004590, -0.003223], |
| | [0.008966, 0.009299, 0.005486], |
| | [-0.016937, -0.008969, -0.014726], |
| | [0.005981, 0.006354, -0.006855], |
| | [0.009750, 0.006882, 0.007736], |
| | [0.001820, 0.004259, 0.012132], |
| | [0.012835, 0.012450, 0.011795], |
| | [0.003041, 0.010194, 0.013934], |
| | [-0.016527, -0.032534, -0.030963], |
| | [-0.015136, -0.007481, -0.009911], |
| | [0.030708, 0.021832, 0.025773], |
| | [-0.008353, -0.012020, -0.008660], |
| | [0.018777, 0.017951, 0.006013], |
| | [-0.006846, -0.006453, -0.005759], |
| | [0.017944, 0.016239, 0.017806], |
| | [-0.009166, -0.004829, 0.002145], |
| | [0.011764, 0.010028, 0.008942], |
| | [0.015022, -0.016713, -0.031551], |
| | [-0.103677, -0.102297, -0.093770], |
| | [-0.006865, -0.003216, -0.002682], |
| | [-0.007705, 0.001121, -0.012102], |
| | [0.015788, -0.003327, 0.006230], |
| | [-0.005562, -0.009712, -0.008889], |
| | [0.006411, 0.011945, 0.014182], |
| | [-0.003523, -0.003832, -0.008597], |
| | [-0.002705, -0.007006, -0.002440], |
| | [0.010826, 0.021793, 0.019520], |
| | [0.021403, 0.017133, 0.011349], |
| | [-0.020997, -0.001073, -0.013768], |
| | [-0.004439, 0.005850, 0.001262], |
| | [0.008814, 0.004013, 0.004906], |
| | [0.008196, 0.005846, 0.007751], |
| | [0.000102, 0.001182, 0.005504], |
| | [-0.007416, -0.009051, -0.006597], |
| | [0.039224, 0.052929, 0.060699], |
| | [0.006937, 0.009651, 0.004330], |
| | [-0.013241, -0.008414, -0.010154], |
| | [-0.021549, 0.012296, 0.043766], |
| | [0.009432, -0.007242, 0.003204], |
| | [-0.038648, -0.035801, -0.032508], |
| | [0.009745, 0.013935, 0.012653], |
| | [-0.024403, -0.018949, -0.025871], |
| | [-0.002547, -0.010403, -0.004686], |
| | [-0.000516, 0.000137, 0.000241], |
| | [-0.001571, -0.000162, -0.010227], |
| | [-0.023109, -0.024477, -0.019911], |
| | [-0.003461, -0.001731, -0.004959], |
| | [0.001869, 0.002194, 0.001378], |
| | [-0.011800, -0.001712, -0.004228], |
| | [-0.019423, -0.002752, 0.007851], |
| | [-0.000403, -0.011006, -0.011421], |
| | [0.000448, -0.002081, -0.002759], |
| | [-0.004389, -0.005971, -0.034043], |
| | [-0.013471, -0.013140, -0.013874], |
| | ] |
| | latent_rgb_factors_bias = [-0.048359, -0.119311, -0.188382] |
| | |
| |
|
| | elif model_family == "ltxv": |
| | latent_channels = 128 |
| | latent_dimensions = 3 |
| |
|
| | latent_rgb_factors = [ |
| | [ 1.1202e-02, -6.3815e-04, -1.0021e-02], |
| | [ 8.6031e-02, 6.5813e-02, 9.5409e-04], |
| | [-1.2576e-02, -7.5734e-03, -4.0528e-03], |
| | [ 9.4063e-03, -2.1688e-03, 2.6093e-03], |
| | [ 3.7636e-03, 1.2765e-02, 9.1548e-03], |
| | [ 2.1024e-02, -5.2973e-03, 3.4373e-03], |
| | [-8.8896e-03, -1.9703e-02, -1.8761e-02], |
| | [-1.3160e-02, -1.0523e-02, 1.9709e-03], |
| | [-1.5152e-03, -6.9891e-03, -7.5810e-03], |
| | [-1.7247e-03, 4.6560e-04, -3.3839e-03], |
| | [ 1.3617e-02, 4.7077e-03, -2.0045e-03], |
| | [ 1.0256e-02, 7.7318e-03, 1.3948e-02], |
| | [-1.6108e-02, -6.2151e-03, 1.1561e-03], |
| | [ 7.3407e-03, 1.5628e-02, 4.4865e-04], |
| | [ 9.5357e-04, -2.9518e-03, -1.4760e-02], |
| | [ 1.9143e-02, 1.0868e-02, 1.2264e-02], |
| | [ 4.4575e-03, 3.6682e-05, -6.8508e-03], |
| | [-4.5681e-04, 3.2570e-03, 7.7929e-03], |
| | [ 3.3902e-02, 3.3405e-02, 3.7454e-02], |
| | [-2.3001e-02, -2.4877e-03, -3.1033e-03], |
| | [ 5.0265e-02, 3.8841e-02, 3.3539e-02], |
| | [-4.1018e-03, -1.1095e-03, 1.5859e-03], |
| | [-1.2689e-01, -1.3107e-01, -2.1005e-01], |
| | [ 2.6276e-02, 1.4189e-02, -3.5963e-03], |
| | [-4.8679e-03, 8.8486e-03, 7.8029e-03], |
| | [-1.6610e-03, -4.8597e-03, -5.2060e-03], |
| | [-2.1010e-03, 2.3610e-03, 9.3796e-03], |
| | [-2.2482e-02, -2.1305e-02, -1.5087e-02], |
| | [-1.5753e-02, -1.0646e-02, -6.5083e-03], |
| | [-4.6975e-03, 5.0288e-03, -6.7390e-03], |
| | [ 1.1951e-02, 2.0712e-02, 1.6191e-02], |
| | [-6.3704e-03, -8.4827e-03, -9.5483e-03], |
| | [ 7.2610e-03, -9.9326e-03, -2.2978e-02], |
| | [-9.1904e-04, 6.2882e-03, 9.5720e-03], |
| | [-3.7178e-02, -3.7123e-02, -5.6713e-02], |
| | [-1.3373e-01, -1.0720e-01, -5.3801e-02], |
| | [-5.3702e-03, 8.1256e-03, 8.8397e-03], |
| | [-1.5247e-01, -2.1437e-01, -2.1843e-01], |
| | [ 3.1441e-02, 7.0335e-03, -9.7541e-03], |
| | [ 2.1528e-03, -8.9817e-03, -2.1023e-02], |
| | [ 3.8461e-03, -5.8957e-03, -1.5014e-02], |
| | [-4.3470e-03, -1.2940e-02, -1.5972e-02], |
| | [-5.4781e-03, -1.0842e-02, -3.0204e-03], |
| | [-6.5347e-03, 3.0806e-03, -1.0163e-02], |
| | [-5.0414e-03, -7.1503e-03, -8.9686e-04], |
| | [-8.5851e-03, -2.4351e-03, 1.0674e-03], |
| | [-9.0016e-03, -9.6493e-03, 1.5692e-03], |
| | [ 5.0914e-03, 1.2099e-02, 1.9968e-02], |
| | [ 1.3758e-02, 1.1669e-02, 8.1958e-03], |
| | [-1.0518e-02, -1.1575e-02, -4.1307e-03], |
| | [-2.8410e-02, -3.1266e-02, -2.2149e-02], |
| | [ 2.9336e-03, 3.6511e-02, 1.8717e-02], |
| | [-1.6703e-02, -1.6696e-02, -4.4529e-03], |
| | [ 4.8818e-02, 4.0063e-02, 8.7410e-03], |
| | [-1.5066e-02, -5.7328e-04, 2.9785e-03], |
| | [-1.7613e-02, -8.1034e-03, 1.3086e-02], |
| | [-9.2633e-03, 1.0803e-02, -6.3489e-03], |
| | [ 3.0851e-03, 4.7750e-04, 1.2347e-02], |
| | [-2.2785e-02, -2.3043e-02, -2.6005e-02], |
| | [-2.4787e-02, -1.5389e-02, -2.2104e-02], |
| | [-2.3572e-02, 1.0544e-03, 1.2361e-02], |
| | [-7.8915e-03, -1.2271e-03, -6.0968e-03], |
| | [-1.1478e-02, -1.2543e-03, 6.2679e-03], |
| | [-5.4229e-02, 2.6644e-02, 6.3394e-03], |
| | [ 4.4216e-03, -7.3338e-03, -1.0464e-02], |
| | [-4.5013e-03, 1.6082e-03, 1.4420e-02], |
| | [ 1.3673e-02, 8.8877e-03, 4.1253e-03], |
| | [-1.0145e-02, 9.0072e-03, 1.5695e-02], |
| | [-5.6234e-03, 1.1847e-03, 8.1261e-03], |
| | [-3.7171e-03, -5.3538e-03, 1.2590e-03], |
| | [ 2.9476e-02, 2.1424e-02, 3.0424e-02], |
| | [-3.4925e-02, -2.4340e-02, -2.5316e-02], |
| | [-3.4127e-02, -2.2406e-02, -1.0589e-02], |
| | [-1.7342e-02, -1.3249e-02, -1.0719e-02], |
| | [-2.1478e-03, -8.6051e-03, -2.9878e-03], |
| | [ 1.2089e-03, -4.2391e-03, -6.8569e-03], |
| | [ 9.0411e-04, -6.6886e-03, -6.7547e-05], |
| | [ 1.6048e-02, -1.0057e-02, -2.8929e-02], |
| | [ 1.2290e-03, 1.0163e-02, 1.8861e-02], |
| | [ 1.7264e-02, 2.7257e-04, 1.3785e-02], |
| | [-1.3482e-02, -3.6427e-03, 6.7481e-04], |
| | [ 4.6782e-03, -5.2423e-03, 2.4467e-03], |
| | [-5.9113e-03, -6.2244e-03, -1.8162e-03], |
| | [ 1.5496e-02, 1.4582e-02, 1.9514e-03], |
| | [ 7.4958e-03, 1.5886e-03, -8.2305e-03], |
| | [ 1.9086e-02, 1.6360e-03, -3.9674e-03], |
| | [-5.7021e-03, -2.7307e-03, -4.1066e-03], |
| | [ 1.7450e-03, 1.4602e-02, 2.5794e-02], |
| | [-8.2788e-04, 2.2902e-03, 4.5161e-03], |
| | [ 1.1632e-02, 8.9193e-03, -7.2813e-03], |
| | [ 7.5721e-03, 2.6784e-03, 1.1393e-02], |
| | [ 5.1939e-03, 3.6903e-03, 1.4049e-02], |
| | [-1.8383e-02, -2.2529e-02, -2.4477e-02], |
| | [ 5.8842e-04, -5.7874e-03, -1.4770e-02], |
| | [-1.6125e-02, -8.6101e-03, -1.4533e-02], |
| | [ 2.0540e-02, 2.0729e-02, 6.4338e-03], |
| | [ 3.3587e-03, -1.1226e-02, -1.6444e-02], |
| | [-1.4742e-03, -1.0489e-02, 1.7097e-03], |
| | [ 2.8130e-02, 2.3546e-02, 3.2791e-02], |
| | [-1.8532e-02, -1.2842e-02, -8.7756e-03], |
| | [-8.0533e-03, -1.0771e-02, -1.7536e-02], |
| | [-3.9009e-03, 1.6150e-02, 3.3359e-02], |
| | [-7.4554e-03, -1.4154e-02, -6.1910e-03], |
| | [ 3.4734e-03, -1.1370e-02, -1.0581e-02], |
| | [ 1.1476e-02, 3.9281e-03, 2.8231e-03], |
| | [ 7.1639e-03, -1.4741e-03, -3.8066e-03], |
| | [ 2.2250e-03, -8.7552e-03, -9.5719e-03], |
| | [ 2.4146e-02, 2.1696e-02, 2.8056e-02], |
| | [-5.4365e-03, -2.4291e-02, -1.7802e-02], |
| | [ 7.4263e-03, 1.0510e-02, 1.2705e-02], |
| | [ 6.2669e-03, 6.2658e-03, 1.9211e-02], |
| | [ 1.6378e-02, 9.4933e-03, 6.6971e-03], |
| | [ 1.7173e-02, 2.3601e-02, 2.3296e-02], |
| | [-1.4568e-02, -9.8279e-03, -1.1556e-02], |
| | [ 1.4431e-02, 1.4430e-02, 6.6362e-03], |
| | [-6.8230e-03, 1.8863e-02, 1.4555e-02], |
| | [ 6.1156e-03, 3.4700e-03, -2.6662e-03], |
| | [-2.6983e-03, -5.9402e-03, -9.2276e-03], |
| | [ 1.0235e-02, 7.4173e-03, -7.6243e-03], |
| | [-1.3255e-02, 1.9322e-02, -9.2153e-04], |
| | [ 2.4222e-03, -4.8039e-03, -1.5759e-02], |
| | [ 2.6244e-02, 2.5951e-02, 2.0249e-02], |
| | [ 1.5711e-02, 1.8498e-02, 2.7407e-03], |
| | [-2.1714e-03, 4.7214e-03, -2.2443e-02], |
| | [-7.4747e-03, 7.4166e-03, 1.4430e-02], |
| | [-8.3906e-03, -7.9776e-03, 9.7927e-03], |
| | [ 3.8321e-02, 9.6622e-03, -1.9268e-02], |
| | [-1.4605e-02, -6.7032e-03, 3.9675e-03] |
| | ] |
| | latent_rgb_factors_bias = [-0.0571, -0.1657, -0.2512] |
| |
|
| | elif model_family == "hunyuan": |
| | if sub_family == "hunyuan1.5": |
| | |
| | latent_rgb_factors = [ |
| | [0.031652, -0.021645, -0.006670], |
| | [-0.015073, -0.012539, -0.013397], |
| | [0.004683, 0.014850, 0.000513], |
| | [-0.012478, -0.008691, 0.017885], |
| | [0.046208, 0.069739, 0.062120], |
| | [0.006511, 0.006715, -0.049985], |
| | [0.025992, 0.025690, 0.015917], |
| | [0.013518, 0.002994, 0.019133], |
| | [0.035577, -0.028987, -0.005096], |
| | [-0.006799, -0.000338, 0.004619], |
| | [-0.003666, -0.004885, -0.008002], |
| | [-0.032795, -0.022699, -0.001088], |
| | [0.039256, 0.051807, 0.054762], |
| | [0.011453, 0.010160, -0.039282], |
| | [0.020320, 0.015219, 0.013882], |
| | [-0.014381, -0.029567, -0.014211], |
| | [0.036559, -0.021804, -0.009747], |
| | [-0.010908, -0.018426, -0.017130], |
| | [0.011040, 0.028708, 0.020362], |
| | [-0.007885, -0.009209, 0.015580], |
| | [0.042282, 0.056767, 0.048389], |
| | [0.018727, 0.026872, -0.022482], |
| | [0.016667, 0.023371, 0.022561], |
| | [-0.016530, -0.026502, -0.005821], |
| | [0.056069, -0.012952, 0.000057], |
| | [-0.001732, 0.000677, -0.002098], |
| | [0.015616, 0.027786, 0.024835], |
| | [-0.002916, 0.003586, 0.016754], |
| | [0.046222, 0.049560, 0.044939], |
| | [0.004627, 0.002370, -0.040525], |
| | [-0.001537, 0.004215, 0.007437], |
| | [0.001337, 0.001811, 0.022117], |
| | ] |
| | latent_rgb_factors_bias = [0.106222, 0.041891, -0.005932] |
| | else: |
| | latent_channels = 16 |
| | latent_dimensions = 3 |
| | scale_factor = 0.476986 |
| | latent_rgb_factors = [ |
| | [-0.0395, -0.0331, 0.0445], |
| | [ 0.0696, 0.0795, 0.0518], |
| | [ 0.0135, -0.0945, -0.0282], |
| | [ 0.0108, -0.0250, -0.0765], |
| | [-0.0209, 0.0032, 0.0224], |
| | [-0.0804, -0.0254, -0.0639], |
| | [-0.0991, 0.0271, -0.0669], |
| | [-0.0646, -0.0422, -0.0400], |
| | [-0.0696, -0.0595, -0.0894], |
| | [-0.0799, -0.0208, -0.0375], |
| | [ 0.1166, 0.1627, 0.0962], |
| | [ 0.1165, 0.0432, 0.0407], |
| | [-0.2315, -0.1920, -0.1355], |
| | [-0.0270, 0.0401, -0.0821], |
| | [-0.0616, -0.0997, -0.0727], |
| | [ 0.0249, -0.0469, -0.1703] |
| | ] |
| |
|
| | latent_rgb_factors_bias = [ 0.0259, -0.0192, -0.0761] |
| | else: |
| | latent_rgb_factors_bias = latent_rgb_factors = None |
| | return latent_rgb_factors, latent_rgb_factors_bias |
| |
|
