qwerty45-uiop commited on
Commit
b0043bb
·
verified ·
1 Parent(s): 851786a

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -158,7 +158,7 @@ LLM_DATABASE = {
158
  { name: "BLOOM-560M", size: "1.1GB", description: "BigScience's small multilingual" }
159
  ],
160
  "code": [
161
- { name: "CodeT5-small", size: "0.3GB", description: "Compact code generation model" },
162
  { name: "Replit-code-v1-3B", size: "1.2GB", description: "Code completion specialist" },
163
  { name: "UnixCoder-base", size: "0.5GB", description: "Microsoft's code understanding model" },
164
  { name: "CodeBERT-base", size: "0.5GB", description: "Bimodal pre-trained model for programming" },
@@ -177,7 +177,7 @@ LLM_DATABASE = {
177
  },
178
  "low": { # 3-4GB
179
  "general": [
180
- { name: "Phi-1.5", size: "2.8GB", description: "Microsoft's efficient reasoning model" },
181
  { name: "Gemma-2B", size: "1.4GB", description: "Google's compact foundation model" },
182
  { name: "OpenLLaMA-3B", size: "2.1GB", description: "Open source LLaMA reproduction" },
183
  { name: "RedPajama-3B", size: "2.0GB", description: "Together AI's open model" },
 
158
  { name: "BLOOM-560M", size: "1.1GB", description: "BigScience's small multilingual" }
159
  ],
160
  "code": [
161
+ { name: "CodeT5-small", size: "0.3GB", description: "Compact code generation model" },
162
  { name: "Replit-code-v1-3B", size: "1.2GB", description: "Code completion specialist" },
163
  { name: "UnixCoder-base", size: "0.5GB", description: "Microsoft's code understanding model" },
164
  { name: "CodeBERT-base", size: "0.5GB", description: "Bimodal pre-trained model for programming" },
 
177
  },
178
  "low": { # 3-4GB
179
  "general": [
180
+ { name: "Phi-1.5", size: "2.8GB", description: "Microsoft's efficient reasoning model" },
181
  { name: "Gemma-2B", size: "1.4GB", description: "Google's compact foundation model" },
182
  { name: "OpenLLaMA-3B", size: "2.1GB", description: "Open source LLaMA reproduction" },
183
  { name: "RedPajama-3B", size: "2.0GB", description: "Together AI's open model" },