groupuser/triton-python-backend-model
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update 1/model.py

Browse Source
This commit is contained in:
cheetahadmin 2025-09-16 07:39:21 +00:00
parent a36b2b6beb
commit 038d8845d9
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
1/model.py
View File

@ -9,17 +9,21 @@ class TritonPythonModel:
Triton Python Model 클래스. Triton Python Model 클래스.
""" """
def initialize(self, args): def initialize(self, args):
""" print("TritonPythonModel: initialize() called.")
모델이 로드될 때 호출됩니다. self.model_config = json.loads(args['model_config'])
"""
print("TritonPythonModel: initialize() called.") # 출력 설정에서 데이터 타입 정보를 가져옴
self.model_config = json.loads(args['model_config']) output_config = pb_utils.get_output_config_by_name(
self.model_config, "OUTPUT")
output_config = pb_utils.get_output_config_by_name(
self.model_config, "OUTPUT") # Triton 데이터 타입 문자열을 NumPy 데이터 타입으로 직접 변환
self.output_dtype = pb_utils.triton_string_to_np_dtype( # 'BYTES'는 np.object_ 타입에 해당함
output_config['data_type']) if output_config['data_type'] == 'TYPE_STRING':
self.output_dtype = np.object_
else:
# 다른 데이터 타입에 대한 처리 로직 추가 가능
self.output_dtype = pb_utils.triton_string_to_np_dtype(output_config['data_type'])
def execute(self, requests): def execute(self, requests):
""" """