diff --git a/unsloth/device_type.py b/unsloth/device_type.py
index 68038de679..0f924bfdfd 100644
--- a/unsloth/device_type.py
+++ b/unsloth/device_type.py
@@ -24,6 +24,7 @@
 
 import torch
 import functools
+import inspect
 from unsloth_zoo.utils import Version