Dependency package upgrade

pyproject.toml

@@ -28,11 +28,11 @@ dependencies = [
     "multidict==6.0.4",
     "urllib3<2",
     "sentencepiece==0.2.0",
-    "onnx==1.16.0",
-    "onnxruntime==1.16.3",
+    "onnx==1.18.0",
+    "onnxruntime==1.22",
     "numpy==1.26.4",
-    "protobuf==3.20.2",
-    "onnxscript==0.1.0.dev20240327",
+    "protobuf==6.31.0",
+    "onnxscript==0.2.5",
     "pillow===10.4.0",
     "sympy",
     "tensorboard",

tests/peft/test_peft_onnx_transforms.py

@@ -46,18 +46,19 @@ def test_adapter_weights_to_inputs_transform():
 
     out_onnx, transformed = AdapterWeightsToInputsTransform.apply(test_onnx, adapter_name=adapter_name)
     assert transformed
+
     assert (
         onnx.printer.to_text(out_onnx)
         == textwrap.dedent("""
     <
        ir_version: 8,
        opset_import: ["" : 17]
     >
-    test_adapter_weights (float[n,32] input, float[32,32] layer1.weight, float[32,32] layer2.weight) => (float[n,32] output, float[32,32] layer1.weight_RetainedState, float[32,32] layer2.weight_RetainedState) {
-       layer1output = MatMul (input, layer1.weight)
-       output = MatMul (layer1output, layer2.weight)
-       layer1.weight_RetainedState = Identity (layer1.weight)
-       layer2.weight_RetainedState = Identity (layer2.weight)
+    test_adapter_weights (float[n,32] input, float[32,32] "layer1.weight", float[32,32] "layer2.weight") => (float[n,32] output, float[32,32] "layer1.weight_RetainedState", float[32,32] "layer2.weight_RetainedState") {
+       layer1output = MatMul (input, "layer1.weight")
+       output = MatMul (layer1output, "layer2.weight")
+       ["layer1.weight_identity"] "layer1.weight_RetainedState" = Identity ("layer1.weight")
+       ["layer2.weight_identity"] "layer2.weight_RetainedState" = Identity ("layer2.weight")
     }
     """).strip()
     )