Slightly improve the fast previews for flux by adding a bias.

This commit is contained in:
comfyanonymous 2024-10-06 17:55:46 -04:00
parent 2c641e64ad
commit 7d29fbf74b

View File

@ -147,23 +147,24 @@ class Flux(SD3):
self.scale_factor = 0.3611 self.scale_factor = 0.3611
self.shift_factor = 0.1159 self.shift_factor = 0.1159
self.latent_rgb_factors =[ self.latent_rgb_factors =[
[-0.0404, 0.0159, 0.0609], [-0.0346, 0.0244, 0.0681],
[ 0.0043, 0.0298, 0.0850], [ 0.0034, 0.0210, 0.0687],
[ 0.0328, -0.0749, -0.0503], [ 0.0275, -0.0668, -0.0433],
[-0.0245, 0.0085, 0.0549], [-0.0174, 0.0160, 0.0617],
[ 0.0966, 0.0894, 0.0530], [ 0.0859, 0.0721, 0.0329],
[ 0.0035, 0.0399, 0.0123], [ 0.0004, 0.0383, 0.0115],
[ 0.0583, 0.1184, 0.1262], [ 0.0405, 0.0861, 0.0915],
[-0.0191, -0.0206, -0.0306], [-0.0236, -0.0185, -0.0259],
[-0.0324, 0.0055, 0.1001], [-0.0245, 0.0250, 0.1180],
[ 0.0955, 0.0659, -0.0545], [ 0.1008, 0.0755, -0.0421],
[-0.0504, 0.0231, -0.0013], [-0.0515, 0.0201, 0.0011],
[ 0.0500, -0.0008, -0.0088], [ 0.0428, -0.0012, -0.0036],
[ 0.0982, 0.0941, 0.0976], [ 0.0817, 0.0765, 0.0749],
[-0.1233, -0.0280, -0.0897], [-0.1264, -0.0522, -0.1103],
[-0.0005, -0.0530, -0.0020], [-0.0280, -0.0881, -0.0499],
[-0.1273, -0.0932, -0.0680] [-0.1262, -0.0982, -0.0778]
] ]
self.latent_rgb_factors_bias = [-0.0329, -0.0718, -0.0851]
self.taesd_decoder_name = "taef1_decoder" self.taesd_decoder_name = "taef1_decoder"
def process_in(self, latent): def process_in(self, latent):