This commit is contained in:
Galunid 2023-10-31 03:42:55 +01:00
parent 6f6856c6ea
commit b9c664ab2f

View file

@ -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):