ClearML/clearml-serving
1
0
Fork 0
mirror of https://github.com/clearml/clearml-serving synced 2025-02-12 07:25:04 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add str type to Triton type conversion

Browse Source
This commit is contained in:
allegroai 2023-09-23 17:36:28 +03:00
parent 58d826e427
commit 083635c803
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml_serving/engines/triton/triton_helper.py
View File

@ -422,6 +422,8 @@ class TritonHelper(object):
return "FP32" return "FP32"
elif np_dtype == np.float64: elif np_dtype == np.float64:
return "FP64" return "FP64"
elif np_dtype == str:
return "STRING"
elif np_dtype == np.object_ or np_dtype.type == np.bytes_: elif np_dtype == np.object_ or np_dtype.type == np.bytes_:
return "BYTES" return "BYTES"
return None return None