From cb6ece9a1857bbbc8dbc0341d403e8d8950ce78e Mon Sep 17 00:00:00 2001 From: silveroxides Date: Sun, 6 Apr 2025 09:52:32 +0200 Subject: [PATCH] set modelType.FLOW, will cause beta scheduler to work properly --- comfy/model_base.py | 2 +- comfy/supported_models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/model_base.py b/comfy/model_base.py index 2a430362..756a479c 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -1059,7 +1059,7 @@ class Hunyuan3Dv2(BaseModel): return out class Chroma(BaseModel): - def __init__(self, model_config, model_type=ModelType.FLUX, device=None): + def __init__(self, model_config, model_type=ModelType.FLOW, device=None): super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.chroma.model.Chroma) def concat_cond(self, **kwargs): diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 0fe97cb4..9b9c45a2 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -1045,7 +1045,7 @@ class Chroma(supported_models_base.BASE): 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.FLUX, device=device) + out = model_base.Chroma(self, model_type=model_base.ModelType.FLOW, device=device) return out def clip_target(self, state_dict={}):