Change chroma to use Flux shift. (#7961)

Author: comfyanonymous

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):