From 6dc2264d5e8a3f17776d0217e1d6fef554052389 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 6 May 2025 08:59:19 -0400
Subject: [PATCH] Change chroma to use Flux shift.

---
 comfy/model_base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_base.py b/comfy/model_base.py
index 3d33086d8fe..045df131745 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1111,7 +1111,7 @@ def extra_conds(self, **kwargs):
         return out
 
 class Chroma(Flux):
-    def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
+    def __init__(self, model_config, model_type=ModelType.FLUX, device=None):
         super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.chroma.model.Chroma)
 
     def extra_conds(self, **kwargs):