 oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
před 2 roky |
|
oobabooga
|
0f6708c471
Sort the imports
|
před 2 roky |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
před 2 roky |
|
oobabooga
|
f871971de1
Trying to get the chat to work
|
před 2 roky |
|
oobabooga
|
ebd698905c
Add streaming to RWKV
|
před 2 roky |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
před 2 roky |
|
oobabooga
|
ebc64a408c
RWKV support prototype
|
před 2 roky |
|
oobabooga
|
6e843a11d6
Fix FlexGen in chat mode
|
před 2 roky |
|
oobabooga
|
fa58fd5559
Proper way to free the cuda cache
|
před 2 roky |
|
oobabooga
|
700311ce40
Empty the cuda cache at model.generate()
|
před 2 roky |
|
oobabooga
|
78ad55641b
Remove duplicate max_new_tokens parameter
|
před 2 roky |
|
oobabooga
|
65326b545a
Move all gradio elements to shared (so that extensions can use them)
|
před 2 roky |
|
oobabooga
|
9ae063e42b
Fix softprompts when deepspeed is active (#112)
|
před 2 roky |
|
oobabooga
|
7224343a70
Improve the imports
|
před 2 roky |
|
oobabooga
|
1dacd34165
Further refactor
|
před 2 roky |
|
oobabooga
|
ce7feb3641
Further refactor
|
před 2 roky |
