From 170be0ebc8917c223003bec93eff2bf29d03c3b3 Mon Sep 17 00:00:00 2001
From: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
Date: Wed, 7 Jun 2023 04:35:15 +0000
Subject: [PATCH] fix(cli): make sure make_tag to respect config
 trust_remote_code

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
---
 src/openllm/cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/openllm/cli.py b/src/openllm/cli.py
index d4b69a25..e7cdb6e2 100644
--- a/src/openllm/cli.py
+++ b/src/openllm/cli.py
@@ -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":