diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index fc37c1ae0..685f545fb 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -166,7 +166,7 @@ class Model: return RefactModel if model_architecture == "PersimmonForCausalLM": return PersimmonModel - if model_architecture == "StableLMEpochForCausalLM": + if model_architecture in ("StableLMEpochForCausalLM", "LlavaStableLMEpochForCausalLM"): return StableLMModel return Model