From 4ae306664d647f188e26140c09c87c91afc58ec9 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Sun, 26 Oct 2025 23:26:51 -0700
Subject: [PATCH] Update fp8.py

---
 unsloth/kernels/fp8.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/unsloth/kernels/fp8.py b/unsloth/kernels/fp8.py
index 74f28dfaff..9b6673a37c 100644
--- a/unsloth/kernels/fp8.py
+++ b/unsloth/kernels/fp8.py
@@ -23,19 +23,19 @@ torch_matmul = torch.matmul
 
 try:
     from transformers.integrations.finegrained_fp8 import FP8Linear
-except ImportError:
+except:
     FP8Linear = None
     logger.log("Unsloth: FP8 models need importing FP8Linear from `transformers.integrations.finegrained_fp8` but we don't see it.")
 
 try:
     from transformers.integrations.fbgemm_fp8 import FbgemmFp8Linear
-except ImportError:
+except:
     FbgemmFp8Linear = None
     logger.log("Unsloth: FP8 models need importing FbgemmFP8Linear from `transformers.integrations.fbgemm_fp8` but we don't see it.")
 
 try:
     from fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm import triton_quantize_fp8_block
-except ImportError:
+except:
     triton_quantize_fp8_block = None
     logger.log("Unsloth: Could not find fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm.triton_quantize_fp8_block")
 
@@ -43,7 +43,7 @@ try:
     from torchao.prototype.blockwise_fp8_inference.blockwise_quantization import (
         blockwise_fp8_gemm as torchao_blockwise_gemm,
     )
-except ImportError:
+except:
     torchao_blockwise_gemm = None
     logger.log("Unsloth: Could not find torchao.prototype.blockwise_fp8_inference.blockwise_quantization.blockwise_fp8_gemm")