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

Correct misindicated model VRAM requirements

Browse Source
This commit is contained in:
Gnome Ann
2021-09-20 18:49:17 -04:00
parent b89a988ee4
commit bbf2bd4026
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
aiserver.py
View File

@ -46,11 +46,11 @@ class colors:
modellist = [
["Custom Neo (GPT-Neo / Converted GPT-J)", "NeoCustom", ""],
["Custom GPT-2 (eg CloverEdition)", "GPT2Custom", ""],
["GPT Neo 1.3B", "EleutherAI/gpt-neo-1.3B", "4GB"],
["GPT Neo 2.7B", "EleutherAI/gpt-neo-2.7B", "8GB"],
["GPT-2", "gpt2", "600MB"],
["GPT-2 Med", "gpt2-medium", "1GB"],
["GPT-2 Large", "gpt2-large", "8GB"],
["GPT Neo 1.3B", "EleutherAI/gpt-neo-1.3B", "8GB"],
["GPT Neo 2.7B", "EleutherAI/gpt-neo-2.7B", "16GB"],
["GPT-2", "gpt2", "1GB"],
["GPT-2 Med", "gpt2-medium", "2GB"],
["GPT-2 Large", "gpt2-large", "4GB"],
["GPT-2 XL", "gpt2-xl", "8GB"],
["InferKit API (requires API key)", "InferKit", ""],
["Google Colab", "Colab", ""],