 oobabooga
|
7b1b1d8e39
Make the code more like PEP8
|
hace 2 años |
|
oobabooga
|
09b0a3aafb
Add repetition_penalty
|
hace 2 años |
|
oobabooga
|
4578e88ffd
Stop the bot from talking for you in chat mode
|
hace 2 años |
|
oobabooga
|
77294b27dd
Use str(Path) instead of os.path.abspath(Path)
|
hace 2 años |
|
oobabooga
|
78901d522b
Remove unused imports
|
hace 2 años |
|
oobabooga
|
0bd5430988
Use 'with' statement to better handle streaming memory
|
hace 2 años |
|
oobabooga
|
0e16c0bacb
Remove redeclaration of a function
|
hace 2 años |
|
oobabooga
|
19a34941ed
Add proper streaming to RWKV
|
hace 2 años |
|
oobabooga
|
8660227e1b
Add top_k to RWKV
|
hace 2 años |
|
oobabooga
|
153dfeb4dd
Add --rwkv-cuda-on parameter, bump rwkv version
|
hace 2 años |
|
oobabooga
|
6904a507c6
Change some parameters
|
hace 2 años |
|
oobabooga
|
5bed607b77
Increase repetition frequency/penalty for RWKV
|
hace 2 años |
|
oobabooga
|
e91f4bc25a
Add RWKV tokenizer
|
hace 2 años |
|
oobabooga
|
ff9f649c0c
Remove some unused imports
|
hace 2 años |
|
oobabooga
|
a2a3e8f797
Add --rwkv-strategy parameter
|
hace 2 años |
|
oobabooga
|
449116a510
Fix RWKV paths on Windows (attempt)
|
hace 2 años |
|
oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
hace 2 años |
|
oobabooga
|
2f16ce309a
Rename a variable
|
hace 2 años |
|
oobabooga
|
9e9cfc4b31
Parameters
|
hace 2 años |
|
oobabooga
|
0f6708c471
Sort the imports
|
hace 2 años |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
hace 2 años |
|
oobabooga
|
659bb76722
Add RWKVModel class
|
hace 2 años |
|
oobabooga
|
9c86a1cd4a
Add RWKV pip package
|
hace 2 años |
|
oobabooga
|
67ee7bead7
Add cpu, bf16 options
|
hace 2 años |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
hace 2 años |
