From 989db341493144fd3c3a49b7fca786eff9a5d9ef Mon Sep 17 00:00:00 2001 From: Galunid Date: Sun, 29 Oct 2023 02:05:28 +0100 Subject: [PATCH] Missing variable --- model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model.py b/model.py index 9291666c5..66b350f65 100644 --- a/model.py +++ b/model.py @@ -424,6 +424,7 @@ class BaichuanModel(Model): # Collect tensors from generator object model_kv = dict(self.get_tensors()) block_count = self.hparams["num_hidden_layers"] + head_count = self.hparams["num_attention_heads"] tensor_map = gguf.get_tensor_name_map(self.model_arch, block_count) for i in range(block_count):