File size: 211 Bytes
8969f81
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
{
    "models_to_load": [
        "gpt2/small",
        "gpt2/medium",
        "gpt2/large",
        "gpt2/arxiv-nlp",

        "gpt",
        "xlnet",
        "distilgpt2/small"
    ],
    "gpu_per_worker": 1
}