Transformers-Fine-Tuner / model_selector.py
Canstralian's picture
Update model_selector.py
4a6742e verified
raw
history blame
370 Bytes
from transformers import AutoModelForSequenceClassification
def get_model_list():
# List of popular pre-trained models
models = [
"Qwen/Qwen2.5-Coder-32B-Instruct",
"microsoft/Phi-3-mini-4k-instruct",
"bigcode/starcoder2-3b",
"codellama/CodeLlama-34b-Instruct-hf"
"microsoft/Phi-3.5-mini-instruct"
]
return models