grouped-sampling-demo / available_models.py
yonikremer's picture
added llama and alpaca
eeec271
raw
history blame contribute delete
278 Bytes
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",
)