From 75734397a75371fa477ec1edb7be7e2f803659b3 Mon Sep 17 00:00:00 2001 From: Baris Demir Date: Thu, 30 Jul 2026 12:52:13 +0100 Subject: [PATCH] Arm backend: Fix torchtune NF4Tensor import CommitlyExtended fails while importing torchtune.modules.common_utils because the currently pinned torchtune revision imports NF4Tensor from torchao.quantization. Bump torchtune to a revision that imports NF4Tensor from torchao.dtypes.nf4tensor, matching the torchao package installed in CI. Signed-off-by: Baris Demir Change-Id: I16d8fa47825337fe5575a3cfd4c1c545f3df3d71 --- requirements-examples.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements-examples.txt b/requirements-examples.txt index 1f36b7c6108..f1579edcb14 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 @ git+https://github.com/pytorch/torchtune.git@3c1872ace149f03ef4ffec765d3f0a5fd0399497 +torchtune @ git+https://github.com/pytorch/torchtune.git@6f2aa7254458145f99d7004cbd6ebc8e53a06404 transformers == 5.0.0rc1