groupuser/triton-gemma-3-1b-it
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modify model.py

Browse Source
This commit is contained in:
groupuser 2025-05-20 15:15:36 +09:00
parent f7f4d8c5f2
commit d7323f2065
1 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
1/model.py
View File

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