We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 69f99db commit ffed03dCopy full SHA for ffed03d
src/jaqpot_api_client/models/model_type.py
@@ -28,6 +28,7 @@ class ModelType(str, Enum):
28
allowed enum values
29
"""
30
SKLEARN_ONNX = 'SKLEARN_ONNX'
31
+ TORCH_ONNX = 'TORCH_ONNX'
32
TORCH_SEQUENCE_ONNX = 'TORCH_SEQUENCE_ONNX'
33
TORCH_GEOMETRIC_ONNX = 'TORCH_GEOMETRIC_ONNX'
34
TORCHSCRIPT = 'TORCHSCRIPT'
0 commit comments