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