| AVAILABLE_MODELS = ( | |
| "facebook/opt-iml-max-1.3b", | |
| "facebook/opt-iml-max-30b", | |
| "gpt2", | |
| "chavinlo/alpaca-13b", | |
| "decapoda-research/llama-7b-hf", | |
| "decapoda-research/llama-13b-hf", | |
| "decapoda-research/llama-30b-hf", | |
| "decapoda-research/llama-65b-hf", | |
| ) | |
| AVAILABLE_MODELS = ( | |
| "facebook/opt-iml-max-1.3b", | |
| "facebook/opt-iml-max-30b", | |
| "gpt2", | |
| "chavinlo/alpaca-13b", | |
| "decapoda-research/llama-7b-hf", | |
| "decapoda-research/llama-13b-hf", | |
| "decapoda-research/llama-30b-hf", | |
| "decapoda-research/llama-65b-hf", | |
| ) | |