ColabWan / shared /RGB_factors.py
1ripon1's picture
Upload folder using huggingface_hub
7344bef verified
Raw
History Blame Contribute Delete
27.5 kB
# thanks Comfyui for the rgb factors (https://github.com/comfyanonymous/ComfyUI/blob/master/comfy/latent_formats.py) (if not explictly computed by DeepBeepMeep as indicated)
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]
# latent_rgb_factors_bias = [0.0259, -0.0192, -0.0761]
elif model_family =="flux":
if sub_family =="flux2":
# computed by your very own Deep Beep Meep
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":
# computed by your very own Deep Beep Meep
if model_type == "ltx2_22B":
ltx2_22b_rgb_factors = [
[-0.002637, -0.022435, -0.014525],
[-0.000040, 0.002258, -0.002773],
[0.017013, 0.019457, 0.019020],
[0.003543, -0.003552, -0.007818],
[-0.009536, -0.003874, 0.004383],
[0.004008, -0.006709, -0.004319],
[-0.036751, -0.043466, -0.048329],
[-0.096575, -0.064235, -0.062143],
[0.011298, 0.014851, 0.011954],
[-0.010023, -0.009218, -0.020513],
[-0.008272, -0.008638, -0.006964],
[-0.020323, -0.015916, -0.013125],
[-0.000465, -0.000317, -0.008137],
[-0.016116, -0.018386, -0.020579],
[-0.006978, -0.000909, 0.007134],
[0.016408, 0.007682, 0.002734],
[-0.005839, -0.000587, 0.012418],
[-0.004253, 0.000283, 0.000204],
[0.011091, 0.012431, 0.020362],
[-0.010974, -0.013548, -0.007926],
[0.008863, -0.002148, -0.014016],
[-0.001360, 0.001420, 0.000509],
[-0.011228, -0.010231, 0.003795],
[-0.003145, -0.012330, -0.000278],
[-0.004863, 0.006857, 0.019429],
[-0.001873, -0.009766, -0.000230],
[-0.012370, -0.005718, -0.009533],
[-0.012286, -0.001246, 0.006019],
[0.004865, -0.002002, -0.004165],
[-0.004346, -0.005383, -0.004504],
[-0.002492, 0.004697, 0.002858],
[-0.002394, 0.000685, 0.005236],
[0.007604, 0.007468, 0.009615],
[-0.000106, -0.003317, -0.000438],
[0.024229, 0.025679, 0.025601],
[0.008970, 0.005995, 0.006451],
[0.000735, 0.001644, 0.001438],
[-0.000723, -0.003181, -0.005735],
[-0.007293, -0.006387, -0.004542],
[0.006823, 0.004482, 0.005263],
[-0.003291, 0.002319, 0.000601],
[-0.019049, -0.022004, -0.015870],
[-0.005866, -0.006257, -0.004923],
[-0.008179, -0.005068, -0.012506],
[0.016921, 0.010924, 0.008665],
[0.016727, 0.006363, 0.019086],
[0.007689, 0.009421, 0.006278],
[0.000311, 0.004294, 0.002340],
[-0.008876, -0.010579, -0.007904],
[-0.004933, -0.004116, -0.003897],
[-0.007960, -0.007913, -0.005858],
[0.033702, 0.029980, 0.023054],
[0.003681, 0.000736, 0.001179],
[0.011194, 0.009676, 0.008584],
[-0.006748, -0.008025, -0.012712],
[0.003629, 0.008801, 0.024646],
[-0.005553, -0.003534, 0.000868],
[0.018373, 0.026107, 0.048515],
[-0.018188, -0.016738, -0.011824],
[-0.005569, -0.002895, -0.000267],
[0.024878, 0.022724, 0.020622],
[-0.003171, -0.003368, -0.007122],
[0.016151, 0.019179, 0.016925],
[-0.020461, -0.016825, 0.005923],
[-0.010058, 0.008644, 0.006714],
[-0.030223, -0.028451, -0.028098],
[-0.012519, -0.010286, -0.010658],
[-0.002638, -0.002010, 0.005868],
[0.009401, 0.010221, 0.013040],
[-0.004573, -0.005399, -0.002189],
[0.005468, 0.008426, 0.007985],
[0.004175, 0.001259, 0.002328],
[-0.008226, 0.031092, 0.034302],
[-0.002707, 0.000329, -0.000929],
[-0.017879, -0.011023, -0.008185],
[-0.154098, -0.149095, -0.114443],
[-0.011193, -0.010340, -0.010992],
[0.008415, 0.009909, 0.006711],
[0.010234, 0.006162, -0.000474],
[-0.003913, -0.004917, 0.004581],
[0.009658, -0.004839, -0.002031],
[0.000503, -0.001236, -0.002399],
[0.045968, 0.025765, 0.022633],
[-0.015219, -0.026419, -0.025316],
[-0.008027, -0.001555, 0.004266],
[0.002060, -0.000470, -0.000015],
[0.017509, 0.020194, 0.006590],
[0.030655, 0.015133, -0.001706],
[0.005769, 0.002032, -0.001164],
[-0.011608, -0.002177, -0.000190],
[-0.006229, -0.005444, -0.003577],
[0.000107, 0.005813, -0.001181],
[-0.014333, -0.017095, 0.004329],
[0.020339, 0.016295, 0.015720],
[0.019904, 0.013839, 0.007771],
[0.008964, 0.009021, 0.008781],
[-0.004733, 0.014352, 0.008170],
[-0.082997, -0.146227, -0.177141],
[-0.004424, 0.004261, -0.002953],
[0.002149, 0.005867, 0.009992],
[0.009515, 0.016390, 0.019565],
[0.002039, -0.008968, -0.009913],
[-0.007031, -0.003982, 0.000093],
[0.068442, 0.075373, 0.051207],
[0.011794, 0.007765, -0.000760],
[0.001684, -0.003199, 0.009644],
[-0.003405, -0.004217, -0.006650],
[-0.015646, -0.006160, -0.000201],
[-0.060917, -0.058964, -0.050211],
[-0.001947, -0.003689, 0.002990],
[0.018446, 0.000534, -0.040772],
[-0.001858, 0.008965, 0.015008],
[-0.045379, -0.037718, -0.035154],
[-0.000329, 0.007689, 0.000036],
[-0.016536, -0.021012, -0.004650],
[0.018388, 0.023352, 0.016103],
[-0.086596, -0.074310, -0.078423],
[0.009143, 0.008548, 0.004895],
[0.002656, 0.002964, -0.000803],
[-0.008501, -0.008217, -0.001661],
[0.006445, -0.003139, -0.008694],
[0.023101, 0.035471, 0.040335],
[0.005615, 0.007732, 0.007626],
[-0.017656, -0.014044, -0.009303],
[-0.004839, -0.003358, -0.000664],
[-0.007193, -0.005399, -0.008577],
[-0.031276, -0.034591, -0.036400],
[-0.008987, -0.015887, -0.013977]
]
ltx2_22b_rgb_factors_bias = [-0.219173, -0.292845, -0.314850]
return ltx2_22b_rgb_factors, ltx2_22b_rgb_factors_bias
# LTX2_RGB_FACTORS_START
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]
# LTX2_RGB_FACTORS_END
elif model_family == "ltxv":
# computed by your very own Deep Beep Meep
latent_channels = 128
latent_dimensions = 3
latent_rgb_factors = [
[0.035582, 0.025241, 0.019059],
[-0.011088, -0.004560, -0.005371],
[0.007245, -0.014716, -0.001559],
[0.020083, 0.002775, -0.002997],
[0.008828, 0.023831, 0.011037],
[0.008272, 0.012256, 0.003322],
[-0.015408, -0.019657, -0.013805],
[-0.012167, -0.010744, -0.040915],
[-0.000306, 0.002370, 0.005182],
[0.014412, 0.007726, -0.001534],
[-0.008717, -0.006287, 0.000956],
[-0.006908, -0.004871, 0.001747],
[0.013476, 0.019110, 0.016148],
[0.006494, -0.002371, -0.006191],
[0.034034, 0.018516, 0.023265],
[-0.001736, 0.003750, 0.013429],
[-0.001662, -0.011720, -0.010040],
[-0.013817, -0.002600, -0.006292],
[0.034908, 0.034661, 0.041437],
[-0.003544, -0.027082, -0.021179],
[0.014436, 0.008519, 0.014859],
[0.000617, -0.001874, -0.004648],
[-0.034923, -0.020644, -0.003953],
[-0.003528, -0.001817, 0.000409],
[0.007435, 0.016008, 0.024745],
[0.008246, -0.015555, -0.030704],
[-0.029761, -0.025874, -0.016636],
[-0.001359, -0.006993, -0.002812],
[0.006954, 0.006185, 0.006284],
[-0.004145, 0.002543, 0.000443],
[0.026397, 0.011885, 0.011944],
[0.011959, 0.006476, 0.004567],
[0.011958, 0.003278, 0.005483],
[0.000880, 0.006825, -0.000239],
[0.003066, 0.002158, 0.001152],
[-0.000697, -0.000912, 0.003622],
[0.015179, 0.019700, 0.021853],
[-0.015571, -0.030838, -0.036792],
[0.000705, -0.005045, -0.000780],
[0.000676, 0.001296, -0.003527],
[-0.005033, 0.002671, -0.001989],
[0.014950, 0.013809, 0.011505],
[0.004975, 0.004733, 0.017948],
[-0.000495, -0.009110, -0.037703],
[-0.054368, -0.052336, -0.044666],
[-0.003123, -0.015530, -0.011048],
[-0.006658, -0.000507, 0.001055],
[0.066555, 0.083396, 0.061881],
[-0.023717, -0.018802, -0.022264],
[-0.001998, -0.006345, -0.004872],
[-0.005142, -0.007320, -0.019123],
[0.008872, 0.007865, 0.004287],
[0.002672, -0.002818, -0.002681],
[0.011259, -0.006732, -0.001830],
[0.003965, 0.002713, 0.013869],
[0.035881, 0.021339, 0.019710],
[-0.020165, -0.015751, 0.000726],
[-0.016233, -0.004130, -0.008461],
[-0.019038, -0.019087, -0.016846],
[0.010145, 0.011604, 0.018285],
[-0.009972, -0.009455, -0.010444],
[0.007420, 0.001248, 0.010543],
[-0.017763, -0.024588, -0.018310],
[-0.005316, -0.003467, -0.001688],
[0.075308, 0.060410, 0.042607],
[0.016281, 0.018724, 0.015634],
[0.003319, 0.002911, -0.012188],
[0.002586, -0.000781, 0.002897],
[0.019921, 0.031291, 0.023804],
[0.000507, 0.000265, 0.006217],
[0.015093, 0.009080, 0.012170],
[-0.027553, -0.002317, 0.008801],
[0.019748, -0.006231, -0.010628],
[-0.000992, 0.014483, 0.006747],
[-0.005176, -0.006966, -0.006231],
[0.009369, 0.009001, 0.006080],
[-0.019502, -0.010986, -0.016263],
[0.009622, 0.011231, -0.003507],
[0.011710, 0.007743, 0.007796],
[0.001709, 0.003981, 0.012354],
[0.013475, 0.012501, 0.013031],
[0.000135, 0.005721, 0.011183],
[-0.017884, -0.031334, -0.028284],
[-0.015754, -0.007742, -0.011511],
[0.030132, 0.022401, 0.024860],
[-0.007276, -0.012084, -0.009154],
[0.019716, 0.018651, 0.006074],
[-0.006242, -0.005683, -0.005029],
[0.019383, 0.016312, 0.017580],
[-0.010341, -0.004952, 0.002420],
[0.011134, 0.009569, 0.008515],
[0.019734, -0.012654, -0.028811],
[-0.099568, -0.099130, -0.092332],
[-0.008051, -0.004986, -0.003922],
[-0.014421, -0.004248, -0.015386],
[0.015564, -0.004664, 0.006721],
[-0.006878, -0.010097, -0.009609],
[0.006329, 0.012266, 0.013791],
[-0.004897, -0.006628, -0.011967],
[-0.006143, -0.009694, -0.002289],
[0.012134, 0.022289, 0.017772],
[0.022571, 0.017612, 0.010859],
[-0.021120, -0.001161, -0.012361],
[-0.005513, 0.005725, 0.002040],
[0.009246, 0.004407, 0.005687],
[0.008619, 0.006166, 0.007369],
[-0.001817, 0.000493, 0.005933],
[-0.009593, -0.011080, -0.008062],
[0.045089, 0.058717, 0.064705],
[0.007387, 0.010615, 0.004683],
[-0.014032, -0.009903, -0.011561],
[-0.017529, 0.012640, 0.041473],
[0.007375, -0.007793, 0.001781],
[-0.042459, -0.038817, -0.034248],
[0.009118, 0.012473, 0.011103],
[-0.024234, -0.019062, -0.029192],
[-0.003629, -0.011813, -0.005117],
[-0.001368, -0.000385, 0.000487],
[-0.001570, -0.000605, -0.012106],
[-0.023129, -0.023683, -0.020173],
[-0.002227, -0.000535, -0.003988],
[0.001467, 0.001789, 0.001113],
[-0.010948, -0.000490, -0.003436],
[-0.018422, -0.001083, 0.008414],
[-0.001576, -0.009604, -0.008839],
[-0.000041, -0.001356, -0.001524],
[-0.004037, -0.006747, -0.034852],
[-0.014760, -0.015261, -0.015292]
]
latent_rgb_factors_bias = [-0.063785, -0.133025, -0.195400]
elif model_family == "hunyuan":
if sub_family == "hunyuan1.5":
# computed by your very own Deep Beep Meep
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