Update model_selector.py
Browse files- model_selector.py +5 -4
model_selector.py
CHANGED
|
@@ -3,9 +3,10 @@ from transformers import AutoModelForSequenceClassification
|
|
| 3 |
def get_model_list():
|
| 4 |
# List of popular pre-trained models
|
| 5 |
models = [
|
| 6 |
-
"
|
| 7 |
-
"
|
| 8 |
-
"
|
| 9 |
-
"
|
|
|
|
| 10 |
]
|
| 11 |
return models
|
|
|
|
| 3 |
def get_model_list():
|
| 4 |
# List of popular pre-trained models
|
| 5 |
models = [
|
| 6 |
+
"Qwen/Qwen2.5-Coder-32B-Instruct",
|
| 7 |
+
"microsoft/Phi-3-mini-4k-instruct",
|
| 8 |
+
"bigcode/starcoder2-3b",
|
| 9 |
+
"codellama/CodeLlama-34b-Instruct-hf"
|
| 10 |
+
"microsoft/Phi-3.5-mini-instruct"
|
| 11 |
]
|
| 12 |
return models
|