diff options
| author | Rose Hogenson <rhogenson@posteo.net> | 2022-02-15 09:02:32 -0800 |
|---|---|---|
| committer | Rose Hogenson <rhogenson@posteo.net> | 2022-02-15 09:02:32 -0800 |
| commit | 37d8729645cfb2eefc92192b1d67b2476b81b3c4 (patch) | |
| tree | 347146d1e530a1665c1a2399fbe4a6fb35c70bc1 /ai_dungeon.py | |
| parent | 4817605a0c52a52dc1b9fcae6774887990eb8a3f (diff) | |
| download | gpt-adventure-37d8729645cfb2eefc92192b1d67b2476b81b3c4.tar.zst | |
Use gpt2 by default.
I'm actually getting way better results from gpt2. Now gpt2 is my new
best friend!
Diffstat (limited to 'ai_dungeon.py')
| -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) |