summaryrefslogtreecommitdiff
path: root/convert-llama-ggmlv3-to-gguf.py
diff options
context:
space:
mode:
Diffstat (limited to 'convert-llama-ggmlv3-to-gguf.py')
-rwxr-xr-xconvert-llama-ggmlv3-to-gguf.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/convert-llama-ggmlv3-to-gguf.py b/convert-llama-ggmlv3-to-gguf.py
index 3f39bc39..08ba0c49 100755
--- a/convert-llama-ggmlv3-to-gguf.py
+++ b/convert-llama-ggmlv3-to-gguf.py
@@ -7,9 +7,13 @@ import struct
import sys
from pathlib import Path
-import gguf
import numpy as np
+import os
+if 'NO_LOCAL_GGUF' not in os.environ:
+ sys.path.insert(1, str(Path(__file__).parent / 'gguf-py' / 'gguf'))
+import gguf
+
# Note: Does not support GGML_QKK_64
QK_K = 256
# Items here are (block size, type size)