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