vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Woops

Browse source
This commit is contained in:
Galunid 2023-10-31 03:42:55 +01:00
parent 6f6856c6ea
commit b9c664ab2f
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
model.py
View file

@ -53,7 +53,8 @@ class Model:
return gguf.MODEL_ARCH.STARCODER return gguf.MODEL_ARCH.STARCODER
if arch == "GPTRefactForCausalLM": if arch == "GPTRefactForCausalLM":
return gguf.MODEL_ARCH.REFACT return gguf.MODEL_ARCH.REFACT
if arch == "" if arch == "PersimmonForCausalLM":
return gguf.MODEL_ARCH.PERSIMMON
raise NotImplementedError(f'Architecture "{arch}" not supported!') raise NotImplementedError(f'Architecture "{arch}" not supported!')
@ -194,6 +195,8 @@ class Model:
return StarCoderModel return StarCoderModel
if model_architecture == "GPTRefactForCausalLM": if model_architecture == "GPTRefactForCausalLM":
return RefactModel return RefactModel
if model_architecture == "PersimmonForCausalLM":
return PersimmonModel
return Model return Model
class StableLMModel(Model): class StableLMModel(Model):