phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Add a --manifest-file option to llama download

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-17 10:08:00 -07:00
parent b8fc4d4dee
commit 5e072d0780
4 changed files with 78 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_toolchain/common/model_utils.py
View file

@ -1,9 +1,13 @@
import os
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from llama_models.datatypes import Model
import os
from .config_dirs import DEFAULT_CHECKPOINT_DIR
def model_local_dir(model: Model) -> str:
return os.path.join(DEFAULT_CHECKPOINT_DIR, model.descriptor())
def model_local_dir(descriptor: str) -> str:
return os.path.join(DEFAULT_CHECKPOINT_DIR, descriptor)