This commit is contained in:
Xi Yan 2024-10-25 10:43:22 -07:00
parent 58c3c45f19
commit 81ed0327f3
11 changed files with 12 additions and 7 deletions

View file

@ -13,14 +13,14 @@ from functools import lru_cache
from pathlib import Path
TEMPLATES_PATH = (
Path(os.path.relpath(__file__)).parent.parent.parent.parent / "distributions"
Path(os.path.relpath(__file__)).parent.parent.parent.parent / "distributions----"
)
# build.yaml templates exist in the llama-stack/distributions while wheel installs llama-stack/llama_stack
# we copied the distributions folder to llama-stack/llama_stack/cli/distributions for wheel builds,
# so we need to check both locations
if not TEMPLATES_PATH.exists():
TEMPLATES_PATH = Path(os.path.relpath(__file__)).parent.parent / "distributions"
TEMPLATES_PATH = Path(os.path.relpath(__file__)).parent.parent.parent / "templates"
@lru_cache()