Add vision model support
This commit is contained in:
parent
e87d709446
commit
9e035cdab7
1 changed files with 1 additions and 1 deletions
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue