From a24695dcc2e8bd34eca2cdc00e93a20bc2704c65 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 2 Jan 2026 03:41:51 -0800
Subject: [PATCH] Update import_fixes.py

---
 unsloth/import_fixes.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/unsloth/import_fixes.py b/unsloth/import_fixes.py
index f388f4ea8d..da0fbc613b 100644
--- a/unsloth/import_fixes.py
+++ b/unsloth/import_fixes.py
@@ -97,6 +97,8 @@ if os.environ.get("UNSLOTH_ENABLE_LOGGING", "0") != "1":
     sys.stderr = HidePrintMessage(sys.stderr)
     # https://github.com/pytorch/FBGEMM/blob/d99cd96490ec4aabac2ee95b1e76ea4dcfcfa628/fbgemm_gpu/experimental/gemm/triton_gemm/utils.py#L43-L52
     sys.stderr.add_filter("TMA benchmarks will be running")
+    # Skipping import of cpp extensions due to incompatible torch version 2.9.0+cu128 for torchao version 0.15.0
+    logging.getLogger("torchao").setLevel(logging.ERROR)
 
 
 # Fix up AttributeError: 'MessageFactory' object has no attribute 'GetPrototype'