Add vision model support

This commit is contained in:
Galunid 2023-11-11 08:20:20 +01:00
parent e87d709446
commit 9e035cdab7

View file

@ -166,7 +166,7 @@ class Model:
return RefactModel return RefactModel
if model_architecture == "PersimmonForCausalLM": if model_architecture == "PersimmonForCausalLM":
return PersimmonModel return PersimmonModel
if model_architecture == "StableLMEpochForCausalLM": if model_architecture in ("StableLMEpochForCausalLM", "LlavaStableLMEpochForCausalLM"):
return StableLMModel return StableLMModel
return Model return Model