add env for LLAMA_STACK_CONFIG_DIR

This commit is contained in:
Xi Yan 2024-09-27 14:11:41 -07:00
parent 43744455d7
commit a3fea59eb5

View file

@ -8,7 +8,7 @@ import os
from pathlib import Path from pathlib import Path
LLAMA_STACK_CONFIG_DIR = Path(os.path.expanduser("~/.llama/")) LLAMA_STACK_CONFIG_DIR = Path(os.getenv("LLAMA_STACK_CONFIG_DIR", os.path.expanduser("~/.llama/")))
DISTRIBS_BASE_DIR = LLAMA_STACK_CONFIG_DIR / "distributions" DISTRIBS_BASE_DIR = LLAMA_STACK_CONFIG_DIR / "distributions"