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

fix mistake

Browse Source
This commit is contained in:
silveroxides 2025-04-08 12:56:02 +02:00
parent e1af413722
commit 3d375a153e
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/supported_models.py
View File

@ -17,6 +17,7 @@ import comfy.text_encoders.hunyuan_video
import comfy.text_encoders.cosmos
import comfy.text_encoders.lumina2
import comfy.text_encoders.wan
import comfy.text_encoders.chroma
from . import supported_models_base
from . import latent_formats
@ -1034,6 +1035,7 @@ class Chroma(supported_models_base.BASE):
}
sampling_settings = {
"multiplier": 1.0,
}
latent_format = comfy.latent_formats.Flux
@ -1042,6 +1044,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, device=device)
return out