Merge pull request #149 from oobabooga/RWKV

Add RWKV support
This commit is contained in:
oobabooga
2023-03-01 16:57:45 -03:00
committed by GitHub
5 changed files with 73 additions and 1 deletions

View File

@@ -5,6 +5,7 @@ tokenizer = None
model_name = ""
soft_prompt_tensor = None
soft_prompt = False
is_RWKV = False
# Chat variables
history = {'internal': [], 'visible': []}