From 216386f3f545e92e2cd19a0cf5de34d83ca72de0 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Thu, 22 Feb 2024 11:27:27 +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 198711f6a..937dfcc56 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -218,7 +218,7 @@ class Model: return BertModel if model_architecture == "NomicBertModel": return NomicBertModel - if model_architecture in "GemmaForCausalLM": + if model_architecture == "GemmaForCausalLM": return GemmaModel return Model