address comment
This commit is contained in:
parent
15f233b9a1
commit
ee5b171250
1 changed files with 2 additions and 0 deletions
|
@ -222,6 +222,8 @@ class Model:
|
|||
return NomicBertModel
|
||||
if model_architecture == "GemmaForCausalLM":
|
||||
return GemmaModel
|
||||
if model_architecture == "Starcoder2ForCausalLM":
|
||||
return Model
|
||||
return Model
|
||||
|
||||
def _is_model_safetensors(self) -> bool:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue