Skip to content

Commit c031312

Browse files
committed
Missing falcon declarations
1 parent 63dd07a commit c031312

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

model.py

+5
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,9 @@ def _get_model_architecture(self):
4747
return gguf.MODEL_ARCH.MPT
4848
if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
4949
return gguf.MODEL_ARCH.BAICHUAN
50+
if arch == "FalconForCausalLM"
51+
return gguf.MODEL_ARCH.FALCON
52+
5053
raise NotImplementedError(f'Architecture "{arch}" not supported!')
5154

5255
def set_vocab(self):
@@ -180,6 +183,8 @@ def from_model_architecture(model_architecture):
180183
return MPTModel
181184
if model_architecture in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
182185
return BaichuanModel
186+
if mode_architecture == "FalconForCausalLM":
187+
return FalconModel
183188
return Model
184189

185190
class StableLMModel(Model):

0 commit comments

Comments
 (0)