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

Falcon HF compatibility

Browse source 
Falcon HF compatibility
This commit is contained in:
John 2023-11-16 23:24:34 +01:00 • committed by GitHub
parent 8da46278e1
commit 8e9396aadd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert-hf-to-gguf.py
View file

@ -193,7 +193,7 @@ class Model:
return gguf.MODEL_ARCH.MPT return gguf.MODEL_ARCH.MPT
if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"): if arch in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
return gguf.MODEL_ARCH.BAICHUAN return gguf.MODEL_ARCH.BAICHUAN
if arch == "FalconForCausalLM": if arch in ("FalconForCausalLM", "RWForCausalLM"):
return gguf.MODEL_ARCH.FALCON return gguf.MODEL_ARCH.FALCON
if arch == "GPTBigCodeForCausalLM": if arch == "GPTBigCodeForCausalLM":
return gguf.MODEL_ARCH.STARCODER return gguf.MODEL_ARCH.STARCODER