rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

OPT models

Browse Source
This commit is contained in:
Henk
2022-05-12 23:55:21 +02:00
parent 376e76f5da
commit e98cc3cb16
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
aiserver.py
View File

@ -79,6 +79,7 @@ mainmenu = [
["Chatbot Models", "chatlist", ""], ["Chatbot Models", "chatlist", ""],
["Untuned GPT-Neo/J", "gptneolist", ""], ["Untuned GPT-Neo/J", "gptneolist", ""],
["Untuned Fairseq Dense", "fsdlist", ""], ["Untuned Fairseq Dense", "fsdlist", ""],
["Untuned OPT", "optlist", ""],
["Untuned XGLM", "xglmlist", ""], ["Untuned XGLM", "xglmlist", ""],
["Untuned GPT2", "gpt2list", ""], ["Untuned GPT2", "gpt2list", ""],
["Online Services", "apilist", ""], ["Online Services", "apilist", ""],
@ -137,6 +138,17 @@ gpt2list = [
["Return to Main Menu", "Return", ""], ["Return to Main Menu", "Return", ""],
] ]
optlist = [
["OPT 30B", "facebook/opt-30b", ""],
["OPT 13B", "facebook/opt-13b", "32GB"],
["OPT 6.7B", "facebook/opt-6.7b", "16GB"],
["OPT 2.7B", "facebook/opt-2.7b", "8GB"],
["OPT 1.3B", "facebook/opt-1.3b", "4GB"],
["OPT 355M", "facebook/opt-350m", "2GB"],
["OPT 125M", "facebook/opt-125m", "1GB"],
["Return to Main Menu", "Return", ""],
]
fsdlist = [ fsdlist = [
["Fairseq Dense 13B", "KoboldAI/fairseq-dense-13B", "32GB"], ["Fairseq Dense 13B", "KoboldAI/fairseq-dense-13B", "32GB"],
["Fairseq Dense 6.7B", "KoboldAI/fairseq-dense-6.7B", "16GB"], ["Fairseq Dense 6.7B", "KoboldAI/fairseq-dense-6.7B", "16GB"],