diff --git a/extensions_built_in/diffusion_models/chroma/chroma_model.py b/extensions_built_in/diffusion_models/chroma/chroma_model.py index 3888909a..9bf3e51e 100644 --- a/extensions_built_in/diffusion_models/chroma/chroma_model.py +++ b/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): diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts index 2c33ad72..71fdc9d8 100644 --- a/ui/src/app/jobs/new/options.ts +++ b/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'],