Duh - add llava in another place

ggml-org · Galunid · Nov 14, 2023 · Oct 11, 2023 · Oct 11, 2023 · Oct 12, 2023
commit 047032d689870ff244eeadc4fbf2a1e28a5c9452
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -201,7 +201,7 @@ def _get_model_architecture(self) -> gguf.MODEL_ARCH:
             return gguf.MODEL_ARCH.REFACT
         if arch == "PersimmonForCausalLM":
             return gguf.MODEL_ARCH.PERSIMMON
-        if arch == "StableLMEpochForCausalLM":
+        if arch in ("StableLMEpochForCausalLM", "LlavaStableLMEpochForCausalLM"):
             return gguf.MODEL_ARCH.STABLELM
 
         raise NotImplementedError(f'Architecture "{arch}" not supported!')