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): 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 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"