298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

set modelType.FLOW, will cause beta scheduler to work properly

Browse Source
This commit is contained in:
silveroxides 2025-04-06 09:52:32 +02:00
parent de3f3ea28b
commit cb6ece9a18
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_base.py
View File

@ -1059,7 +1059,7 @@ class Hunyuan3Dv2(BaseModel):
return out return out
class Chroma(BaseModel): 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) super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.chroma.model.Chroma)
def concat_cond(self, **kwargs): def concat_cond(self, **kwargs):

2
comfy/supported_models.py
View File

@ -1045,7 +1045,7 @@ class Chroma(supported_models_base.BASE):
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32] supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
def get_model(self, state_dict, prefix="", device=None): 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 return out
def clip_target(self, state_dict={}): def clip_target(self, state_dict={}):