print file list at input model directory
This commit is contained in:
parent
941523384f
commit
ddb8145bd1
10
1/model.py
10
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}")
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user