Use gpt2 by default.

I'm actually getting way better results from gpt2. Now gpt2 is my new
best friend!

1 files changed, 1 insertions, 1 deletions

diff --git a/ai_dungeon.py b/ai_dungeon.py
index 9d28bb3..2b976f6 100755
--- a/ai_dungeon.py
+++ b/ai_dungeon.py
@@ -105,7 +105,7 @@ def main() -> None:
     """Run the main game loop."""
     parser = argparse.ArgumentParser("AI dungeon clone")
     parser.add_argument(
-        "--model", default="EleutherAI/gpt-neo-125M", help="Model to use"
+        "--model", default="gpt2", help="Model to use"
     )
     args = parser.parse_args()
     model = load_model(args.model)