{
    "models_to_load": [
        "gpt2/small",
        "gpt2/medium",
        "gpt2/large",
        "gpt2/arxiv-nlp",

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