We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 63dd07a commit c031312Copy full SHA for c031312
model.py
@@ -47,6 +47,9 @@ def _get_model_architecture(self):
47
return gguf.MODEL_ARCH.MPT
48
if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
49
return gguf.MODEL_ARCH.BAICHUAN
50
+ if arch == "FalconForCausalLM"
51
+ return gguf.MODEL_ARCH.FALCON
52
+
53
raise NotImplementedError(f'Architecture "{arch}" not supported!')
54
55
def set_vocab(self):
@@ -180,6 +183,8 @@ def from_model_architecture(model_architecture):
180
183
return MPTModel
181
184
if model_architecture in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
182
185
return BaichuanModel
186
+ if mode_architecture == "FalconForCausalLM":
187
+ return FalconModel
188
return Model
189
190
class StableLMModel(Model):
0 commit comments