From 2c47b8a7ac7c1b94badbdd38f89005dac95d5cff Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 26 Oct 2025 23:29:14 -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 9b6673a37c..0e893a360f 100644 --- a/unsloth/kernels/fp8.py +++ b/unsloth/kernels/fp8.py @@ -25,19 +25,19 @@ try: from transformers.integrations.finegrained_fp8 import FP8Linear except: FP8Linear = None - logger.log("Unsloth: FP8 models need importing FP8Linear from `transformers.integrations.finegrained_fp8` but we don't see it.") + logger.info("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: FbgemmFp8Linear = None - logger.log("Unsloth: FP8 models need importing FbgemmFP8Linear from `transformers.integrations.fbgemm_fp8` but we don't see it.") + logger.info("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: triton_quantize_fp8_block = None - logger.log("Unsloth: Could not find fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm.triton_quantize_fp8_block") + logger.info("Unsloth: Could not find fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm.triton_quantize_fp8_block") try: from torchao.prototype.blockwise_fp8_inference.blockwise_quantization import ( @@ -45,7 +45,7 @@ try: ) except: torchao_blockwise_gemm = None - logger.log("Unsloth: Could not find torchao.prototype.blockwise_fp8_inference.blockwise_quantization.blockwise_fp8_gemm") + logger.info("Unsloth: Could not find torchao.prototype.blockwise_fp8_inference.blockwise_quantization.blockwise_fp8_gemm") @triton.jit