 oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
3 роки тому |
|
oobabooga
|
2f16ce309a
Rename a variable
|
3 роки тому |
|
oobabooga
|
9e9cfc4b31
Parameters
|
3 роки тому |
|
oobabooga
|
0f6708c471
Sort the imports
|
3 роки тому |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
3 роки тому |
|
oobabooga
|
659bb76722
Add RWKVModel class
|
3 роки тому |
|
oobabooga
|
9c86a1cd4a
Add RWKV pip package
|
3 роки тому |
|
oobabooga
|
67ee7bead7
Add cpu, bf16 options
|
3 роки тому |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
3 роки тому |
