Added support for Chroma1 official release. Will still use single file verstion instead of the diffusers version.

extensions_built_in/diffusion_models/chroma/chroma_model.py

@@ -129,6 +129,12 @@ class ChromaModel(BaseModel):
                 repo_id='lodestones/Chroma',
                 filename=f"chroma-unlocked-v{version}.safetensors",
             )
+        elif model_path.startswith("lodestones/Chroma1-"):
+            # will have a file in the repo that is Chroma1-whatever.safetensors
+            model_path = huggingface_hub.hf_hub_download(
+                repo_id=model_path,
+                filename=f"{model_path.split('/')[-1]}.safetensors",
+            )
         else:
             # check if the model path is a local file
             if os.path.exists(model_path):

ui/src/app/jobs/new/options.ts

@@ -106,7 +106,7 @@ export const modelArchs: ModelArch[] = [
     group: 'image',
     defaults: {
       // default updates when [selected, unselected] in the UI
-      'config.process[0].model.name_or_path': ['lodestones/Chroma', defaultNameOrPath],
+      'config.process[0].model.name_or_path': ['lodestones/Chroma1-Base', defaultNameOrPath],
       'config.process[0].model.quantize': [true, false],
       'config.process[0].model.quantize_te': [true, false],
       'config.process[0].sample.sampler': ['flowmatch', 'flowmatch'],