Merge pull request #149 from oobabooga/RWKV

Add RWKV support
2023-03-01 16:57:45 -03:00 · 2023-03-01 16:57:45 -03:00 · f3da6dcc8f
commit f3da6dcc8f
parent a1429d1607 831ac7ed3f
5 changed files with 73 additions and 1 deletions
--- a/modules/shared.py
+++ b/modules/shared.py
@ -5,6 +5,7 @@ tokenizer = None
 model_name = ""
 soft_prompt_tensor = None
 soft_prompt = False
+is_RWKV = False

 # Chat variables
 history = {'internal': [], 'visible': []}