mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

RWKV support prototype

Browse source
This commit is contained in:
oobabooga 2023-02-27 23:03:35 -03:00
parent 021bd55886
commit ebc64a408c
3 changed files with 42 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/shared.py
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': []}