pyrox.dev
1diff --git a/unsloth_zoo/__init__.py b/unsloth_zoo/__init__.py
2index 2332907..2eed5e9 100644
3--- a/unsloth_zoo/__init__.py
4+++ b/unsloth_zoo/__init__.py
5@@ -64,8 +64,6 @@ pass
6
7
8 from importlib.util import find_spec
9-if find_spec("unsloth") is None:
10- raise ImportError("Please install Unsloth via `pip install unsloth`!")
11 pass
12 del find_spec
13
14@@ -75,12 +73,13 @@ def get_device_type():
15 return "cuda"
16 elif hasattr(torch, "xpu") and torch.xpu.is_available():
17 return "xpu"
18+ else:
19+ # Allow import during tests
20+ return None
21 raise NotImplementedError("Unsloth currently only works on NVIDIA GPUs and Intel GPUs.")
22 pass
23 DEVICE_TYPE : str = get_device_type()
24
25-if not ("UNSLOTH_IS_PRESENT" in os.environ):
26- raise ImportError("Please install Unsloth via `pip install unsloth`!")
27 pass
28
29 try: