diff --git a/clearml_serving/engines/triton/triton_helper.py b/clearml_serving/engines/triton/triton_helper.py index 533d06a..ff606cb 100644 --- a/clearml_serving/engines/triton/triton_helper.py +++ b/clearml_serving/engines/triton/triton_helper.py @@ -422,6 +422,8 @@ class TritonHelper(object): return "FP32" elif np_dtype == np.float64: return "FP64" + elif np_dtype == str: + return "STRING" elif np_dtype == np.object_ or np_dtype.type == np.bytes_: return "BYTES" return None