From e167ebcb22226ae1f7adb6ae6bb2bf111d3332e0 Mon Sep 17 00:00:00 2001 From: Galunid Date: Sun, 22 Oct 2023 11:08:19 +0200 Subject: [PATCH] Fix formatting in gguf.py --- gguf-py/gguf/gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gguf-py/gguf/gguf.py b/gguf-py/gguf/gguf.py index 0c44c3e94..235fff7d9 100644 --- a/gguf-py/gguf/gguf.py +++ b/gguf-py/gguf/gguf.py @@ -89,7 +89,7 @@ class MODEL_ARCH(IntEnum): REFACT : int = auto() BERT : int = auto() BLOOM : int = auto() - STABLELM : int = auto() + STABLELM : int = auto() class MODEL_TENSOR(IntEnum): @@ -129,7 +129,7 @@ MODEL_ARCH_NAMES: dict[MODEL_ARCH, str] = { MODEL_ARCH.REFACT: "refact", MODEL_ARCH.BERT: "bert", MODEL_ARCH.BLOOM: "bloom", - MODEL_ARCH.STABLELM: "stablelm", + MODEL_ARCH.STABLELM: "stablelm", } TENSOR_NAMES: dict[MODEL_TENSOR, str] = {