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

Check that the model is found before use. (#182)

Browse source
This commit is contained in:
AshleyT3 2024-10-03 23:24:47 -07:00 • committed by GitHub
parent f913b57397
commit 734f59d3b8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/cli/download.py
View file

@ -158,12 +158,11 @@ def run_download_cmd(args: argparse.Namespace, parser: argparse.ArgumentParser):
info = prompt_guard_download_info() info = prompt_guard_download_info()
else: else:
model = resolve_model(args.model_id) model = resolve_model(args.model_id)
if model is None:
parser.error(f"Model {args.model_id} not found")
return
info = llama_meta_net_info(model) info = llama_meta_net_info(model)
if model is None:
parser.error(f"Model {args.model_id} not found")
return
if args.source == "huggingface": if args.source == "huggingface":
_hf_download(model, args.hf_token, args.ignore_patterns, parser) _hf_download(model, args.hf_token, args.ignore_patterns, parser)
else: else: