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

This commit is contained in:
Jaret Burkett
2025-08-23 09:06:28 -06:00
parent 0d6d027248
commit e1fd411665
2 changed files with 7 additions and 1 deletions

View File

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