diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 35b6de15ef..ee9647c7c6 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -21,6 +21,7 @@ from transformers.models.llama.modeling_llama import logger
 from transformers import AutoTokenizer
 from platform import system as platform_system
 platform_system = platform_system()
+import math
 
 __version__ = "2024.1"