diff options
| -rwxr-xr-x | ai_dungeon.py | 2 |
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) |
