summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xai_dungeon.py2
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)