diff --git a/gguf-py/gguf/__init__.py b/gguf-py/gguf/__init__.py
index 718ea71eb..40b77128e 100644
--- a/gguf-py/gguf/__init__.py
+++ b/gguf-py/gguf/__init__.py
@@ -1 +1 @@
-from .gguf import GGUFWriter
+from .gguf import GGUFWriter, get_tensor_name_map, MODEL_ARCH