From 047032d689870ff244eeadc4fbf2a1e28a5c9452 Mon Sep 17 00:00:00 2001 From: Galunid Date: Sun, 12 Nov 2023 03:48:51 +0100 Subject: [PATCH] Duh - add llava in another place --- 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 685f545fb..e7db75912 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -201,7 +201,7 @@ class Model: 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!')