commit 9b9b3bafbba820701fe1457fffb6ee2f7489e855 Author: jaeheon Date: Tue Sep 30 10:28:43 2025 +0900 python diff --git a/1/model.py b/1/model.py new file mode 100644 index 0000000..d52009d --- /dev/null +++ b/1/model.py @@ -0,0 +1,38 @@ +import triton_python_backend_utils as pb_utils +import numpy as np +import json +import random +import string + +class TritonPythonModel: + def initialize(self, args): + self.logger = pb_utils.Logger + self.logger.log_info(f"'{args["model_name"]}' 모델 초기화 완료") + + def execute(self, requests): + responses = [] + + for request in requests: + input_tensor = pb_utils.get_input_tensor_by_name(request, "INPUT") + + input_data = input_tensor.as_numpy()[0].decode("utf-8") + self.logger.log_info(f"INPUT: {input_data}") + + random_string = ''.join( + random.choice(string.ascii_letters + string.digits) for _ in range(10) + ) + self.logger.log_info(f"OUTPUT: {random_string}") + + output_tensor = pb_utils.Tensor( + "OUTPUT", + np.array([random_string.encode("utf-8")], dtype=np.object_) + ) + + responses.append(pb_utils.InferenceResponse( + output_tensors=[output_tensor] + )) + + return responses + + def finalize(self): + pass diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..a8b322c --- /dev/null +++ b/config.pbtxt @@ -0,0 +1,20 @@ +# Triton backend to use +backend: "python" +max_batch_size: 0 + +input [ + { + name: "INPUT" + data_type: TYPE_STRING + dims: [ -1 ] + } +] + +output [ + { + name: "OUTPUT" + data_type: TYPE_STRING + dims: [ -1 ] + } +] +