latent2rgb matrix for SDXL.

im-stuff · Jul 9, 2023 · d5323d1 · d5323d1
1 parent 0ae81c0
commit d5323d1
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/comfy/latent_formats.py b/comfy/latent_formats.py
@@ -11,10 +11,10 @@ def __init__(self, scale_factor=0.18215):
         self.scale_factor = scale_factor
         self.latent_rgb_factors = [
                     #   R        G        B
-                    [0.298, 0.207, 0.208],  # L1
-                    [0.187, 0.286, 0.173],  # L2
-                    [-0.158, 0.189, 0.264],  # L3
-                    [-0.184, -0.271, -0.473],  # L4
+                    [ 0.3512,  0.2297,  0.3227],
+                    [ 0.3250,  0.4974,  0.2350],
+                    [-0.2829,  0.1762,  0.2721],
+                    [-0.2120, -0.2616, -0.7177]
                 ]
         self.taesd_decoder_name = "taesd_decoder.pth"
 
@@ -23,9 +23,9 @@ def __init__(self):
         self.scale_factor = 0.13025
         self.latent_rgb_factors = [ #TODO: these are the factors for SD1.5, need to estimate new ones for SDXL
                     #   R        G        B
-                    [0.298, 0.207, 0.208],  # L1
-                    [0.187, 0.286, 0.173],  # L2
-                    [-0.158, 0.189, 0.264],  # L3
-                    [-0.184, -0.271, -0.473],  # L4
+                    [ 0.3920,  0.4054,  0.4549],
+                    [-0.2634, -0.0196,  0.0653],
+                    [ 0.0568,  0.1687, -0.0755],
+                    [-0.3112, -0.2359, -0.2076]
                 ]
         self.taesd_decoder_name = "taesdxl_decoder.pth"