diff --git a/1/model.py b/1/model.py index 52f3fb5..bbd0d27 100644 --- a/1/model.py +++ b/1/model.py @@ -22,6 +22,16 @@ class TritonPythonModel: #model_path = f"{model_repository}/{self.model_name}" model_path = "/cheetah/input/model/gemma-3-1b-it" + input_model_path = model_path + + if os.path.exists(input_model_path): + file_list = os.listdir(input_model_path) + self.logger.log_info(f"'{input_model_path}' 디렉토리의 파일 목록:") + for file_name in file_list: + self.logger.log_info(file_name) + else: + self.logger.log_info(f"'{input_model_path}' 디렉토리가 존재하지 않습니다.") + self.logger.log_info(f"model_repository: {model_repository}") self.logger.log_info(f"model_path: {model_path}")