diff --git a/test/unit/test_gguf.py b/test/unit/test_gguf.py index fe690082b0..0b136bcb4d 100644 --- a/test/unit/test_gguf.py +++ b/test/unit/test_gguf.py @@ -1,4 +1,4 @@ -import os, struct, tempfile, unittest +import os, struct, unittest from tinygrad import dtypes, Tensor, fetch, Device from tinygrad.nn.state import ggml_data_to_tensor, gguf_load from tinygrad.device import is_dtype_supported @@ -135,26 +135,22 @@ class TestGGUFGEMV(unittest.TestCase): q_data = q_data.flatten() ref = dequantize(q_data, qtype).reshape(rows, cols) - # build a minimal gguf: header + 1 tensor info + aligned data - with tempfile.NamedTemporaryFile(suffix=".gguf", delete=False) as f: - f.write(struct.pack("<4siqq", b"GGUF", 3, 1, 0)) # magic, version, n_tensors, n_kv - f.write(struct.pack("