From 0cf2e46b1725a5d0d6cb7b177a524026ca00f5a4 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Tue, 6 May 2025 07:39:54 -0400
Subject: [PATCH 1/3] ComfyUI version 0.3.32

---
 comfyui_version.py | 2 +-
 pyproject.toml     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfyui_version.py b/comfyui_version.py
index 8d2068de7fd0..61573aead5c0 100644
--- a/comfyui_version.py
+++ b/comfyui_version.py
@@ -1,3 +1,3 @@
 # This file is automatically generated by the build process when version is
 # updated in pyproject.toml.
-__version__ = "0.3.31"
+__version__ = "0.3.32"
diff --git a/pyproject.toml b/pyproject.toml
index 8b549a0b6f52..878e7c66a9f2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "ComfyUI"
-version = "0.3.31"
+version = "0.3.32"
 readme = "README.md"
 license = { file = "LICENSE" }
 requires-python = ">=3.9"

From a4e679765ec6a7adc6dda326c7bab915e2914bd6 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 6 May 2025 06:00:01 -0700
Subject: [PATCH 2/3] Change chroma to use Flux shift. (#7961)

---
 comfy/model_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 3d33086d8fe1..045df1317458 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1111,7 +1111,7 @@ def extra_conds(self, **kwargs):
         return out
 
 class Chroma(Flux):
-    def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
+    def __init__(self, model_config, model_type=ModelType.FLUX, device=None):
         super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.chroma.model.Chroma)
 
     def extra_conds(self, **kwargs):

From 271c9c5b9eb027c682161ed7848bbf640253f973 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 6 May 2025 06:52:37 -0700
Subject: [PATCH 3/3] Better mem estimation for the LTXV 13B model. (#7963)

---
 comfy/supported_models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index d5210cfac440..a1dea234307d 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -785,6 +785,10 @@ class LTXV(supported_models_base.BASE):
     vae_key_prefix = ["vae."]
     text_encoder_key_prefix = ["text_encoders."]
 
+    def __init__(self, unet_config):
+        super().__init__(unet_config)
+        self.memory_usage_factor = (unet_config.get("cross_attention_dim", 2048) / 2048) * 5.5
+
     def get_model(self, state_dict, prefix="", device=None):
         out = model_base.LTXV(self, device=device)
         return out