Update src/streamlit_app.py
Browse files- 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 |
-
|
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 |
-
|
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" },
|