 oobabooga
|
7bbe32f618
Don't return a value in an iterator function
|
2 年 前 |
|
oobabooga
|
ff9f649c0c
Remove some unused imports
|
2 年 前 |
|
oobabooga
|
955cf431e8
Minor consistency fix
|
2 年 前 |
|
oobabooga
|
831ac7ed3f
Add top_p
|
2 年 前 |
|
oobabooga
|
7c4d5ca8cc
Improve the text generation call a bit
|
2 年 前 |
|
oobabooga
|
0f6708c471
Sort the imports
|
2 年 前 |
|
oobabooga
|
e735806c51
Add a generate() function for RWKV
|
2 年 前 |
|
oobabooga
|
f871971de1
Trying to get the chat to work
|
2 年 前 |
|
oobabooga
|
ebd698905c
Add streaming to RWKV
|
2 年 前 |
|
oobabooga
|
70e522732c
Move RWKV loader into a separate file
|
2 年 前 |
|
oobabooga
|
ebc64a408c
RWKV support prototype
|
2 年 前 |
|
oobabooga
|
6e843a11d6
Fix FlexGen in chat mode
|
2 年 前 |
|
oobabooga
|
fa58fd5559
Proper way to free the cuda cache
|
2 年 前 |
|
oobabooga
|
700311ce40
Empty the cuda cache at model.generate()
|
2 年 前 |
|
oobabooga
|
78ad55641b
Remove duplicate max_new_tokens parameter
|
2 年 前 |
|
oobabooga
|
65326b545a
Move all gradio elements to shared (so that extensions can use them)
|
2 年 前 |
|
oobabooga
|
9ae063e42b
Fix softprompts when deepspeed is active (#112)
|
2 年 前 |
|
oobabooga
|
7224343a70
Improve the imports
|
2 年 前 |
|
oobabooga
|
1dacd34165
Further refactor
|
2 年 前 |
|
oobabooga
|
ce7feb3641
Further refactor
|
2 年 前 |
