From c33bdf397de671cfc863439a8def7994d9516df3 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng Date: Sun, 28 Apr 2024 21:52:33 -0400 Subject: [PATCH] fix improper function signature --- gguf-py/gguf/gguf_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gguf-py/gguf/gguf_manager.py b/gguf-py/gguf/gguf_manager.py index b1e680810..5218f3a80 100644 --- a/gguf-py/gguf/gguf_manager.py +++ b/gguf-py/gguf/gguf_manager.py @@ -188,7 +188,7 @@ class GGUFManager: print("Model has smaller size than the split threshold, not splitting") self.split_style = SplitStyle.NONE - self.split_strategy = SplitStrategy(self.split_style, self.path, self.tensors, self.args, not self.args.large_first_shard) + self.split_strategy = SplitStrategy(self.split_style, self.path, self.tensors, self.args, self.arch) self.total_shards = len(self.split_strategy) # only the first shard needs all the KV data