From 9e035cdab76bbc1d82c38f7912efcbea4794d9c7 Mon Sep 17 00:00:00 2001 From: Galunid Date: Sat, 11 Nov 2023 08:20:20 +0100 Subject: [PATCH] Add vision model support --- 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 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