diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml index 6d6f1827d..fe0975465 100644 --- a/.github/workflows/pythonapp.yml +++ b/.github/workflows/pythonapp.yml @@ -32,7 +32,7 @@ jobs: python-version: ['3.9', '3.10', '3.11'] env: TVM_VERSION_TAG: v0.17.0 - PYTORCH_VERSION: 2.2.0 + PYTORCH_VERSION: 2.4.1 LLVM_VERSION: 18 steps: diff --git a/hummingbird/ml/_topology.py b/hummingbird/ml/_topology.py index 38448ebbb..e7a8fc6af 100644 --- a/hummingbird/ml/_topology.py +++ b/hummingbird/ml/_topology.py @@ -243,7 +243,7 @@ def convert(topology, backend, test_input, device, extra_config={}): if backend == onnx.__name__: onnx_model_name = output_model_name = None - target_opset = 11 + target_opset = 13 # Set optional configuration options for ONNX if any. if constants.ONNX_OUTPUT_MODEL_NAME in extra_config: