diff --git a/.github/workflows/check-release.yml b/.github/workflows/check-release.yml index d2c4744..53ae6fe 100644 --- a/.github/workflows/check-release.yml +++ b/.github/workflows/check-release.yml @@ -16,7 +16,7 @@ jobs: matrix: os: [ubuntu-latest, macOS-latest, windows-latest] python: ['3.11', '3.12'] - transformers: ['4.48.3', '4.52.4', '4.55.3', 'main'] + transformers: ['4.48.3', '4.52.4', '4.55.4', 'main'] torch: ['2.7', '2.8', 'main'] steps: diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 4a0f16c..adc81d8 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -16,7 +16,7 @@ jobs: matrix: os: [ubuntu-latest] python: ['3.10', '3.11', '3.12', '3.13'] - transformers: ['4.48.3', '4.51.3', '4.52.4', '4.53.3', '4.55.3', 'main'] + transformers: ['4.48.3', '4.51.3', '4.52.4', '4.53.3', '4.55.4', 'main'] torch: ['2.8', 'main'] exclude: - python: '3.10' @@ -28,7 +28,7 @@ jobs: - python: '3.10' transformers: '4.53.3' - python: '3.10' - transformers: '4.55.3' + transformers: '4.55.4' - python: '3.11' torch: 'main' - python: '3.11' @@ -36,7 +36,7 @@ jobs: - python: '3.11' transformers: 'main' - python: '3.11' - transformers: '4.55.3' + transformers: '4.55.4' - python: '3.13' torch: '2.8' - python: '3.13' diff --git a/_unittests/ut_torch_models/test_validate_whole_models.py b/_unittests/ut_torch_models/test_validate_whole_models.py index 060ec81..096528a 100644 --- a/_unittests/ut_torch_models/test_validate_whole_models.py +++ b/_unittests/ut_torch_models/test_validate_whole_models.py @@ -116,28 +116,6 @@ def test_g_validate_model_onnx_dynamo_os_ort(self): onnx_filename = data["onnx_filename"] self.assertExists(onnx_filename) - @requires_torch("2.7") - @hide_stdout() - @ignore_warnings(FutureWarning) - @requires_experimental() - def test_h_validate_model_custom_os_ort(self): - mid = "arnir0/Tiny-LLM" - summary, data = validate_model( - mid, - do_run=True, - verbose=10, - exporter="custom", - dump_folder="dump_test/validate_model_custom_os_ort", - patch=True, - stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6.1") else 0, - optimization="default+os_ort", - ) - self.assertIsInstance(summary, dict) - self.assertIsInstance(data, dict) - self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-4) - onnx_filename = data["onnx_filename"] - self.assertExists(onnx_filename) - @requires_torch("2.7") @hide_stdout() @ignore_warnings(FutureWarning)