diff --git a/pyproject.toml b/pyproject.toml
index 648d2ce4e..ee760734c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -21,6 +21,7 @@ requires-python = ">=3.8,<3.11"
 dependencies = [
     "transformers==4.50.0",
     "huggingface-hub==0.27.0",
+    
     "hf_transfer==0.1.9",
     "peft==0.13.2",
     "datasets==2.20.0",
@@ -28,11 +29,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",
diff --git a/tests/peft/test_peft_onnx_transforms.py b/tests/peft/test_peft_onnx_transforms.py
index f8521deb1..0248dae3b 100644
--- a/tests/peft/test_peft_onnx_transforms.py
+++ b/tests/peft/test_peft_onnx_transforms.py
@@ -46,6 +46,7 @@ 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("""
@@ -53,11 +54,11 @@ def test_adapter_weights_to_inputs_transform():
        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()
     )