From 7720c1c1618d362f259b3b14b0e75946373b79b3 Mon Sep 17 00:00:00 2001
From: Anthony Wang
Date: Sun, 17 Jul 2022 10:48:11 -0500
Subject: Update bot.py to use gpt-large

---
 bot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bot.py b/bot.py
index d285471..34eb039 100644
--- a/bot.py
+++ b/bot.py
@@ -18,8 +18,8 @@ parser.add_argument('-m', '--model', default='model',
 args = parser.parse_args()
 
 
-tokenizer = AutoTokenizer.from_pretrained('gpt2-medium')
-model = AutoModelForCausalLM.from_pretrained(args.model, low_cpu_mem_usage=True).to('cuda')
+tokenizer = AutoTokenizer.from_pretrained('gpt2-large')
+model = AutoModelForCausalLM.from_pretrained(args.model).to('cuda')
 
 
 if args.input is None:
-- 
cgit v1.2.3-70-g09d2