From c2e3b0adc3f805d401d745f69c88cc7e3eb5db75 Mon Sep 17 00:00:00 2001 From: Adam Niederer Date: Sun, 11 Jan 2026 13:47:18 -0500 Subject: [PATCH] Accept size-one ndarrays of any dimension from gguf reader (#394) Sometimes gguf will return size-one one-dimensional memmaps for scalar values. I'm not exactly sure when or why, but this will accept any dimension as long as it only contains one value Resolves #384 --- loader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/loader.py b/loader.py index a17c8ab..37ef133 100644 --- a/loader.py +++ b/loader.py @@ -33,7 +33,7 @@ def get_field(reader, field_name, field_type): raise TypeError(f"Bad type for GGUF {field_name} key: expected string, got {field.types!r}") return str(field.parts[field.data[-1]], encoding="utf-8") elif field_type in [int, float, bool]: - return field_type(field.parts[field.data[-1]]) + return field_type(field.parts[field.data[-1]].item()) else: raise TypeError(f"Unknown field type {field_type}") @@ -427,3 +427,4 @@ def gguf_clip_loader(path): else: pass return sd +