From d93fbc81f618dcf2ffc6b5730f49cec480d6b3c7 Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Sun, 18 Aug 2024 04:27:17 +0200
Subject: [PATCH] Test script

---
 tools/read_tensors.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 tools/read_tensors.py

diff --git a/tools/read_tensors.py b/tools/read_tensors.py
new file mode 100644
index 0000000..1bdff02
--- /dev/null
+++ b/tools/read_tensors.py
@@ -0,0 +1,21 @@
+#!/usr/bin/python3
+import os
+import sys
+import gguf
+
+def read_tensors(path):
+    reader = gguf.GGUFReader(path)
+    for tensor in reader.tensors:
+        if tensor.tensor_type == gguf.GGMLQuantizationType.F32:
+            continue
+        print(f"{str(tensor.tensor_type):32}: {tensor.name}")
+
+try:
+    path = sys.argv[1]
+    assert os.path.isfile(path), "Invalid path"
+    print(f"input: {path}")
+except Exception as e:
+    input(f"failed: {e}")
+else:
+    read_tensors(path)
+    input()