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

Restore support for RWForCausalLM

Browse source
This commit is contained in:
Galunid 2023-11-04 20:43:29 +01:00
parent 007be85087
commit 03c9683eb7
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
model.py
View file

@ -153,7 +153,7 @@ class Model:
return MPTModel return MPTModel
if model_architecture in ("BaichuanForCausalLM", "BaiChuanForCausalLM"): if model_architecture in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
return BaichuanModel return BaichuanModel
if model_architecture == "FalconForCausalLM": if model_architecture in ("FalconForCausalLM", "RWForCausalLM"):
return FalconModel return FalconModel
if model_architecture == "GPTBigCodeForCausalLM": if model_architecture == "GPTBigCodeForCausalLM":
return StarCoderModel return StarCoderModel