1 files changed, 1 insertions, 1 deletions
diff --git a/bot.py b/bot.py
index c3d5274..5bcac18 100644
--- a/bot.py
+++ b/bot.py
@@ -19,7 +19,7 @@ args = parser.parse_args()
 
 
 tokenizer = AutoTokenizer.from_pretrained('gpt2-large')
-model = AutoModelForCausalLM.from_pretrained(args.model, torch_dtype=float16).to('cuda')
+model = AutoModelForCausalLM.from_pretrained(args.model).to('cuda')
 
 
 if args.input is None: