 oobabooga
|
33fb6aed74
Minor bug fix
|
2 vuotta sitten |
|
oobabooga
|
ad2970374a
Readability improvements
|
2 vuotta sitten |
|
oobabooga
|
72d539dbff
Better separate the FlexGen case
|
2 vuotta sitten |
|
oobabooga
|
ab50f80542
New text streaming method (much faster)
|
2 vuotta sitten |
|
oobabooga
|
8e89bc596b
Fix encode() for RWKV
|
2 vuotta sitten |
|
oobabooga
|
19a34941ed
Add proper streaming to RWKV
|
2 vuotta sitten |
|
oobabooga
|
8660227e1b
Add top_k to RWKV
|
2 vuotta sitten |
|
oobabooga
|
20bd645f6a
Fix bug in multigpu setups (attempt 3)
|
2 vuotta sitten |
|
oobabooga
|
09a7c36e1b
Minor improvement while running custom models
|
2 vuotta sitten |
|
oobabooga
|
24c4c20391
Fix bug in multigpu setups (attempt #2)
|
2 vuotta sitten |
|
oobabooga
|
d88b7836c6
Fix bug in multigpu setups
|
2 vuotta sitten |
|
oobabooga
|
e91f4bc25a
Add RWKV tokenizer
|
2 vuotta sitten |
|
oobabooga
|
a54b91af77
Improve readability
|
2 vuotta sitten |
|
oobabooga
|
8e706df20e
Fix a memory leak when text streaming is on
|
2 vuotta sitten |
|
oobabooga
|
c33715ad5b
Move towards HF LLaMA implementation
|
2 vuotta sitten |
|
oobabooga
|
c93f1fa99b
Count the tokens more conservatively
|
2 vuotta sitten |
|
oobabooga
|
05e703b4a4
Print the performance information more reliably
|
2 vuotta sitten |
|
oobabooga
|
a345a2acd2
Add a tokenizer placeholder
|
2 vuotta sitten |
|
oobabooga
|
5b354817f6
Make chat minimally work with LLaMA
|
2 vuotta sitten |
|
oobabooga
|
ea5c5eb3da
Add LLaMA support
|
2 vuotta sitten |
|
oobabooga
|
7bbe32f618
Don't return a value in an iterator function
|
2 vuotta sitten |
|
oobabooga
|
ff9f649c0c
Remove some unused imports
|
2 vuotta sitten |
|
oobabooga
|
955cf431e8
Minor consistency fix
|
2 vuotta sitten |
|
oobabooga
|
831ac7ed3f
Add top_p
|
2 vuotta sitten |
|
oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
2 vuotta sitten |
|
oobabooga
|
0f6708c471
Sort the imports
|
2 vuotta sitten |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
2 vuotta sitten |
|
oobabooga
|
f871971de1
Trying to get the chat to work
|
2 vuotta sitten |
|
oobabooga
|
ebd698905c
Add streaming to RWKV
|
2 vuotta sitten |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
2 vuotta sitten |
