 oobabooga
|
ff9f649c0c
Remove some unused imports
|
vor 2 Jahren |
|
oobabooga
|
1a05860ca3
Ensure proper no-streaming with generation_attempts > 1
|
vor 2 Jahren |
|
oobabooga
|
a2a3e8f797
Add --rwkv-strategy parameter
|
vor 2 Jahren |
|
oobabooga
|
99dc95e14e
Minor aesthetic change
|
vor 2 Jahren |
|
oobabooga
|
449116a510
Fix RWKV paths on Windows (attempt)
|
vor 2 Jahren |
|
oobabooga
|
955cf431e8
Minor consistency fix
|
vor 2 Jahren |
|
oobabooga
|
7a9b4407b0
Settle for 0.0.6 for now
|
vor 2 Jahren |
|
oobabooga
|
f351dce032
Keep rwkv up to date
|
vor 2 Jahren |
|
oobabooga
|
f3da6dcc8f
Merge pull request #149 from oobabooga/RWKV
|
vor 2 Jahren |
|
oobabooga
|
831ac7ed3f
Add top_p
|
vor 2 Jahren |
|
oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
vor 2 Jahren |
|
oobabooga
|
2f16ce309a
Rename a variable
|
vor 2 Jahren |
|
oobabooga
|
9e9cfc4b31
Parameters
|
vor 2 Jahren |
|
oobabooga
|
0f6708c471
Sort the imports
|
vor 2 Jahren |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
vor 2 Jahren |
|
oobabooga
|
659bb76722
Add RWKVModel class
|
vor 2 Jahren |
|
oobabooga
|
9c86a1cd4a
Add RWKV pip package
|
vor 2 Jahren |
|
oobabooga
|
6837d4d72a
Load the model by name
|
vor 2 Jahren |
|
oobabooga
|
a1429d1607
Add default extensions to the settings
|
vor 2 Jahren |
|
oobabooga
|
365e1089b3
Move some buttons
|
vor 2 Jahren |
|
oobabooga
|
c472c34818
Change Chiharu's example dialogue definition format
|
vor 2 Jahren |
|
oobabooga
|
19ccb2aaf5
Handle <USER> and <BOT>
|
vor 2 Jahren |
|
oobabooga
|
626da6c731
Handle {{user}} and {{char}} in example dialogue
|
vor 2 Jahren |
|
oobabooga
|
e861e68e38
Move the chat example dialogue to the prompt
|
vor 2 Jahren |
|
oobabooga
|
f871971de1
Trying to get the chat to work
|
vor 2 Jahren |
|
oobabooga
|
67ee7bead7
Add cpu, bf16 options
|
vor 2 Jahren |
|
oobabooga
|
ebd698905c
Add streaming to RWKV
|
vor 2 Jahren |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
vor 2 Jahren |
|
oobabooga
|
ebc64a408c
RWKV support prototype
|
vor 2 Jahren |
|
oobabooga
|
021bd55886
Better format the prompt when generation attempts > 1
|
vor 2 Jahren |
