From 31dffa061b76d388edce2f734dee802d43e27132 Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Fri, 14 Mar 2025 17:37:14 +0100
Subject: [PATCH] Add error message for missing packages

---
 loader.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/loader.py b/loader.py
index 2f690ed..5253092 100644
--- a/loader.py
+++ b/loader.py
@@ -171,7 +171,10 @@ def llama_permute(raw_sd, n_head, n_head_kv):
 def gguf_tokenizer_loader(path, temb_shape):
     # convert gguf tokenizer to spiece
     print(f"Attempting to recreate sentencepiece tokenizer from GGUF file metadata...")
-    from sentencepiece import sentencepiece_model_pb2 as model
+    try:
+        from sentencepiece import sentencepiece_model_pb2 as model
+    except ImportError:
+        raise ImportError("Please make sure sentencepiece and protobuf are installed.\npip install sentencepiece protobuf")
     spm = model.ModelProto()
 
     reader = gguf.GGUFReader(path)