From 37d8729645cfb2eefc92192b1d67b2476b81b3c4 Mon Sep 17 00:00:00 2001 From: Rose Hogenson Date: Tue, 15 Feb 2022 09:02:32 -0800 Subject: Use gpt2 by default. I'm actually getting way better results from gpt2. Now gpt2 is my new best friend! --- ai_dungeon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ai_dungeon.py') 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) -- cgit v1.3.1