Skip to content

Commit 874b262

Browse files
committed
Update fp8.py
1 parent ccdecc9 commit 874b262

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

unsloth/kernels/fp8.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,27 +25,27 @@
2525
from transformers.integrations.finegrained_fp8 import FP8Linear
2626
except:
2727
FP8Linear = None
28-
logger.log("Unsloth: FP8 models need importing FP8Linear from `transformers.integrations.finegrained_fp8` but we don't see it.")
28+
logger.info("Unsloth: FP8 models need importing FP8Linear from `transformers.integrations.finegrained_fp8` but we don't see it.")
2929

3030
try:
3131
from transformers.integrations.fbgemm_fp8 import FbgemmFp8Linear
3232
except:
3333
FbgemmFp8Linear = None
34-
logger.log("Unsloth: FP8 models need importing FbgemmFP8Linear from `transformers.integrations.fbgemm_fp8` but we don't see it.")
34+
logger.info("Unsloth: FP8 models need importing FbgemmFP8Linear from `transformers.integrations.fbgemm_fp8` but we don't see it.")
3535

3636
try:
3737
from fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm import triton_quantize_fp8_block
3838
except:
3939
triton_quantize_fp8_block = None
40-
logger.log("Unsloth: Could not find fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm.triton_quantize_fp8_block")
40+
logger.info("Unsloth: Could not find fbgemm_gpu.experimental.gemm.triton_gemm.fp8_gemm.triton_quantize_fp8_block")
4141

4242
try:
4343
from torchao.prototype.blockwise_fp8_inference.blockwise_quantization import (
4444
blockwise_fp8_gemm as torchao_blockwise_gemm,
4545
)
4646
except:
4747
torchao_blockwise_gemm = None
48-
logger.log("Unsloth: Could not find torchao.prototype.blockwise_fp8_inference.blockwise_quantization.blockwise_fp8_gemm")
48+
logger.info("Unsloth: Could not find torchao.prototype.blockwise_fp8_inference.blockwise_quantization.blockwise_fp8_gemm")
4949

5050

5151
@triton.jit

0 commit comments

Comments
 (0)