diff --git a/1/model.py b/1/model.py index 8c7ade9..7dee306 100644 --- a/1/model.py +++ b/1/model.py @@ -19,18 +19,8 @@ class TritonPythonModel: self.model_name = args["model_name"] model_repository = args["model_repository"] - model_path = f"{model_repository}/{self.model_name}" - #model_path = "/cheetah/input/model/triton-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}' 디렉토리가 존재하지 않습니다.") + #model_path = f"{model_repository}/{self.model_name}" + model_path = "/cheetah/input/model/triton-gemma-3-1b-it" self.logger.log_info(f"model_repository: {model_repository}") self.logger.log_info(f"model_path: {model_path}")