From 869aeaa050c777487f74df093a57d67e9457d912 Mon Sep 17 00:00:00 2001 From: Erik Lundell Date: Fri, 23 Jan 2026 11:20:36 +0100 Subject: [PATCH] Bump torch-ao pin Signed-off-by: Erik Lundell Change-Id: I8ca07acda3b4c48a19da7a057ec21dba10001fd9 --- backends/arm/requirements-arm-models-test.txt | 4 ++-- backends/cortex_m/test/ops/test_add.py | 7 ++----- requirements-examples.txt | 2 +- third-party/ao | 2 +- 4 files changed, 6 insertions(+), 9 deletions(-) diff --git a/backends/arm/requirements-arm-models-test.txt b/backends/arm/requirements-arm-models-test.txt index 238e9d07c9d..90eee4cfbc5 100644 --- a/backends/arm/requirements-arm-models-test.txt +++ b/backends/arm/requirements-arm-models-test.txt @@ -1,6 +1,6 @@ -# Copyright 2025 Arm Limited and/or its affiliates. +# Copyright 2025-2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. -diffusers[torch] == 0.33.1 \ No newline at end of file +diffusers[torch] @ git+https://github.com/huggingface/diffusers.git@a7cb14efbe4b255f4c8721c91a04f0ef40f4a05a diff --git a/backends/cortex_m/test/ops/test_add.py b/backends/cortex_m/test/ops/test_add.py index ad5f276b544..b776ac91002 100644 --- a/backends/cortex_m/test/ops/test_add.py +++ b/backends/cortex_m/test/ops/test_add.py @@ -1,4 +1,4 @@ -# Copyright 2025 Arm Limited and/or its affiliates. +# Copyright 2025-2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -153,10 +153,7 @@ class CortexMAlphaAdd(ModelAlpha): xfails_implementation = { - "alpha": ( - "Expecting kwargs for aten op IR to be empty - alpha arg not supported.", - AssertionError, - ), + "alpha": "Expecting kwargs for aten op IR to be empty - alpha arg not supported.", } xfails_dialect = xfails_implementation | { # Cortex-M quantizer will not quantize additions that require broadcasting diff --git a/requirements-examples.txt b/requirements-examples.txt index 415e4101312..8420c96333e 100644 --- a/requirements-examples.txt +++ b/requirements-examples.txt @@ -3,5 +3,5 @@ datasets == 3.6.0 # 4.0.0 deprecates trust_remote_code and load scripts. For now pin to 3.6.0 timm == 1.0.7 torchsr == 1.0.4 -torchtune >= 0.6.1 +torchtune @ git+https://github.com/pytorch/torchtune.git@44271b570af36cfda8ee20a4479d2652770378c0 transformers == 5.0.0rc1 diff --git a/third-party/ao b/third-party/ao index 9338966da58..28306f08500 160000 --- a/third-party/ao +++ b/third-party/ao @@ -1 +1 @@ -Subproject commit 9338966da58ec44b60f0e0b173cabab08f942ed0 +Subproject commit 28306f085003b892bc0a250c209d80f5d4a5147b