grouped-sampling-demo / available_models.py
yonikremer's picture
added gpt2 to the models list for testing
818ab4f
raw
history blame
100 Bytes
AVAILABLE_MODELS = (
"facebook/opt-iml-max-1.3b",
"facebook/opt-iml-max-30b",
"gpt2",
)