From c95013d1b5a2eb247e212889c57ca36e738b08fe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stanis=C5=82aw=20Szymczyk?= Date: Thu, 2 May 2024 09:53:59 +0200 Subject: [PATCH] Whitespace formatting fixes. --- convert-hf-to-gguf.py | 2 +- gguf-py/gguf/tensor_mapping.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 6f013a1d0..17cdb0fc5 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1516,6 +1516,7 @@ class LlamaModel(Model): if len(experts) > 0: raise ValueError(f"Unprocessed experts: {experts.keys()}") + @Model.register("ArcticForCausalLM") class ArcticModel(Model): model_arch = gguf.MODEL_ARCH.ARCTIC @@ -1629,7 +1630,6 @@ class ArcticModel(Model): raise ValueError(f"Unprocessed experts: {experts.keys()}") - @Model.register("GrokForCausalLM") class GrokModel(Model): model_arch = gguf.MODEL_ARCH.GROK diff --git a/gguf-py/gguf/tensor_mapping.py b/gguf-py/gguf/tensor_mapping.py index 6ecce589c..844a270f4 100644 --- a/gguf-py/gguf/tensor_mapping.py +++ b/gguf-py/gguf/tensor_mapping.py @@ -427,7 +427,7 @@ class TensorNameMap: MODEL_TENSOR.FFN_NORM_EXP: ( "model.layers.{bid}.post_attention_layernorm", ), - }, + }, } mapping: dict[str, tuple[MODEL_TENSOR, str]]