File size: 214 Bytes
39dbdf0
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from transformers import HUGGINGFACE_HUB_NAME, HUGGINGFACE_HUB_MODEL

def get_model_list():
    return [
        "bert-base-uncased",
        "distilbert-base-uncased",
        "roberta-base",
        "gpt2"
    ]