From 7ad7da6af86078895c4f8917fcb733650056f55f Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Thu, 22 Feb 2024 11:27:32 +0200 Subject: [PATCH] Update convert-hf-to-gguf.py Co-authored-by: Aarni Koskela --- convert-hf-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 937dfcc56..a6f5e58d1 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -279,7 +279,7 @@ class Model: return gguf.MODEL_ARCH.BERT if arch == "NomicBertModel": return gguf.MODEL_ARCH.NOMIC_BERT - if arch in "GemmaForCausalLM": + if arch == "GemmaForCausalLM": return gguf.MODEL_ARCH.GEMMA raise NotImplementedError(f'Architecture "{arch}" not supported!')