Community model

First batch, will be more, we will also need to update the other VRAM display's with the changes that have happened. Will happen depending on how the 8-bit stuff goes.
This commit is contained in:
henk717 2021-12-24 23:12:14 +01:00
parent 6952938e88
commit 1d3370c995
1 changed files with 6 additions and 3 deletions

View File

@ -63,9 +63,12 @@ class colors:
modellist = [ modellist = [
["Load a model from its directory", "NeoCustom", ""], ["Load a model from its directory", "NeoCustom", ""],
["Load an old GPT-2 model (eg CloverEdition)", "GPT2Custom", ""], ["Load an old GPT-2 model (eg CloverEdition)", "GPT2Custom", ""],
["GPT-Neo 1.3B", "EleutherAI/gpt-neo-1.3B", "8GB"], ["Skein 6B", "KoboldAI/GPT-J-6B-Skein", "12GB"],
["GPT-Neo 2.7B", "EleutherAI/gpt-neo-2.7B", "16GB"], ["Adventure 6B", "KoboldAI/GPT-J-6B-Adventure", "12GB"],
["GPT-J 6B", "EleutherAI/gpt-j-6B", "24GB"], ["Picard 6B", "KoboldAI/GPT-Neo-2.7B-Picard", "12GB"],
["GPT-Neo 1.3B", "EleutherAI/gpt-neo-1.3B", "3GB"],
["GPT-Neo 2.7B", "EleutherAI/gpt-neo-2.7B", "6GB"],
["GPT-J 6B", "EleutherAI/gpt-j-6B", "12GB"],
["GPT-2", "gpt2", "1GB"], ["GPT-2", "gpt2", "1GB"],
["GPT-2 Med", "gpt2-medium", "2GB"], ["GPT-2 Med", "gpt2-medium", "2GB"],
["GPT-2 Large", "gpt2-large", "4GB"], ["GPT-2 Large", "gpt2-large", "4GB"],