From 03c9683eb7008a1b14dd04ebe8a97640c3cf674d Mon Sep 17 00:00:00 2001
From: Galunid <karolek1231456@gmail.com>
Date: Sat, 4 Nov 2023 20:43:29 +0100
Subject: [PATCH] Restore support for RWForCausalLM

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 2ebd6881e..c8ae1d51a 100644
--- a/model.py
+++ b/model.py
@@ -153,7 +153,7 @@ class Model:
             return MPTModel
         if model_architecture in ("BaichuanForCausalLM", "BaiChuanForCausalLM"):
             return BaichuanModel
-        if model_architecture == "FalconForCausalLM":
+        if model_architecture in ("FalconForCausalLM", "RWForCausalLM"):
             return FalconModel
         if model_architecture == "GPTBigCodeForCausalLM":
             return StarCoderModel