From 5c7866462c05597097d1cc3b89c21b4ca1bcbf10 Mon Sep 17 00:00:00 2001 From: groupuser Date: Wed, 28 May 2025 08:48:25 +0000 Subject: [PATCH] Update 1/model.py --- 1/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/1/model.py b/1/model.py index a097585..8a6d904 100644 --- a/1/model.py +++ b/1/model.py @@ -12,6 +12,8 @@ class TritonPythonModel: """ self.logger = pb_utils.Logger + self.model_config = json.loads(args["model_config"]) + self.model_name = args["model_name"] self.model_path = self._get_config_parameter("model_path") self.enable_inference_trace = self._get_config_parameter("enable_inference_trace") @@ -23,8 +25,6 @@ class TritonPythonModel: #model_repository = args["model_repository"] #model_path = f"{model_repository}/{self.model_name}" - self.model_config = json.loads(args["model_config"]) - # Hugging Face Transformers 라이브러리에서 사전 학습된 토크나이저를 로드합니다. self.tokenizer = AutoTokenizer.from_pretrained(self.model_path) self.tokenizer.pad_token_id = self.tokenizer.eos_token_id