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

update toolchain to work with updated imports from llama_models

Browse source
This commit is contained in:
Ashwin Bharambe 2024-07-30 17:52:57 -07:00
parent 23014ea4d1
commit 1bc81eae7b
3 changed files with 6 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_toolchain/cli/download.py
View file

@ -16,11 +16,8 @@ import httpx
from huggingface_hub import snapshot_download
from huggingface_hub.utils import GatedRepoError, RepositoryNotFoundError
from llama_models.llama3_1.api.datatypes import (
CheckpointQuantizationFormat,
ModelDefinition,
)
from llama_models.llama3_1.api.sku_list import (
from llama_models.datatypes import CheckpointQuantizationFormat, ModelDefinition
from llama_models.sku_list import (
llama3_1_model_list,
llama_meta_folder_path,
llama_meta_pth_size,