From 6f5c252f55877607d984059e7926aa61a0323896 Mon Sep 17 00:00:00 2001 From: groupuser Date: Tue, 29 Apr 2025 13:00:21 +0900 Subject: [PATCH] add log --- 1/model.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/1/model.py b/1/model.py index 54278db..aa79445 100644 --- a/1/model.py +++ b/1/model.py @@ -2,6 +2,7 @@ import triton_python_backend_utils as pb_utils from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig import numpy as np import json +import os class TritonPythonModel: def initialize(self, args): @@ -12,10 +13,17 @@ class TritonPythonModel: """ self.logger = pb_utils.Logger + current_file_path = os.path.abspath(__file__) + self.logger.log_info(f"current_file_path: {current_file_path}") + + self.model_name = args["model_name"] model_repository = args["model_repository"] model_path = f"{model_repository}/{self.model_name}" + self.logger.log_info(f"model_repository: {model_repository}") + self.logger.log_info(f"model_path: {model_path}") + self.model_config = json.loads(args["model_config"]) # Hugging Face Transformers 라이브러리에서 사전 학습된 토크나이저를 로드합니다.