From fda35f37e5b55f616c13167e481183bf64237af7 Mon Sep 17 00:00:00 2001 From: Silver <65376327+silveroxides@users.noreply.github.com> Date: Sat, 22 Mar 2025 16:49:16 +0100 Subject: [PATCH] Add supported_inference_dtypes --- comfy/supported_models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 1520b58c..da5f3abc 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -1027,6 +1027,7 @@ class Chroma(supported_models_base.BASE): } latent_format = comfy.latent_formats.Flux memory_usage_factor = 2.8 + 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)