ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-28 18:21:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added chroma model to the ui. Added logic to easily pull latest, use local, or use a specific version of chroma. Allow ustom name or path in the ui for custom models

Browse Source
This commit is contained in:
Jaret Burkett
2025-05-07 12:06:30 -06:00
parent d9700bdb99
commit 43cb5603ad
11 changed files with 830 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
config/examples/train_lora_chroma_24gb.yaml
View File

@@ -65,7 +65,14 @@ config:
# Download the whichever model you prefer from the Chroma repo
# https://huggingface.co/lodestones/Chroma/tree/main
# point to it here.
name_or_path: "/path/to/chroma/chroma-unlocked-vVERSION.safetensors"
# name_or_path: "/path/to/chroma/chroma-unlocked-vVERSION.safetensors"
# using lodestones/Chroma will automatically use the latest version
name_or_path: "lodestones/Chroma"
# # You can also select a version of Chroma like so
# name_or_path: "lodestones/Chroma/v28"
arch: "chroma"
quantize: true # run 8bit mixed precision
sample: