diff --git a/unsloth/device_type.py b/unsloth/device_type.py
index 68038de679..6a2d859d15 100644
--- a/unsloth/device_type.py
+++ b/unsloth/device_type.py
@@ -25,6 +25,7 @@ __all__ = [
 import torch
 import functools
 from unsloth_zoo.utils import Version
+import inspect
 
 
 @functools.cache