From e1af4137224fd79f12b9eebf68943bd307276a9d Mon Sep 17 00:00:00 2001 From: silveroxides Date: Tue, 8 Apr 2025 12:48:25 +0200 Subject: [PATCH] Adjust memory usage factor and remove unnecessary code --- comfy/supported_models.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 9b9c45a2..c913545f 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -1034,18 +1034,16 @@ class Chroma(supported_models_base.BASE): } sampling_settings = { - "multiplier": 1.0, - "shift": 1.0, } latent_format = comfy.latent_formats.Flux - memory_usage_factor = 1.8 + memory_usage_factor = 3.2 supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32] def get_model(self, state_dict, prefix="", device=None): - out = model_base.Chroma(self, model_type=model_base.ModelType.FLOW, device=device) + out = model_base.Chroma(self, device=device) return out def clip_target(self, state_dict={}):