Woops
This commit is contained in:
parent
6f6856c6ea
commit
b9c664ab2f
1 changed files with 4 additions and 1 deletions
5
model.py
5
model.py
|
@ -53,7 +53,8 @@ class Model:
|
|||
return gguf.MODEL_ARCH.STARCODER
|
||||
if arch == "GPTRefactForCausalLM":
|
||||
return gguf.MODEL_ARCH.REFACT
|
||||
if arch == ""
|
||||
if arch == "PersimmonForCausalLM":
|
||||
return gguf.MODEL_ARCH.PERSIMMON
|
||||
|
||||
raise NotImplementedError(f'Architecture "{arch}" not supported!')
|
||||
|
||||
|
@ -194,6 +195,8 @@ class Model:
|
|||
return StarCoderModel
|
||||
if model_architecture == "GPTRefactForCausalLM":
|
||||
return RefactModel
|
||||
if model_architecture == "PersimmonForCausalLM":
|
||||
return PersimmonModel
|
||||
return Model
|
||||
|
||||
class StableLMModel(Model):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue