mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-01-23 15:01:32 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(cli): make sure make_tag to respect config trust_remote_code

Browse Source 
Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
aarnphm-ec2-dev
2023-06-07 04:35:15 +00:00
parent ce7143060e
commit 170be0ebc8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/openllm/cli.py
View File

@@ -684,7 +684,7 @@ def cli_factory() -> click.Group:
else:
model = openllm.AutoLLM.for_model(model_name, pretrained=pretrained, llm_config=config)
tag = model.make_tag()
tag = model.make_tag(trust_remote_code=config.__openllm_trust_remote_code__)
if len(bentoml.models.list(tag)) == 0:
if output == "pretty":