available_models.py

AVAILABLE_MODELS = (
    "decapoda-research/llama-30b-hf",
    "decapoda-research/llama-13b-hf",
    "decapoda-research/llama-7b-hf",
    "gpt2",
)