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() )