From d86ced8a0eead191081c231da84e3756668b7c2e Mon Sep 17 00:00:00 2001
From: AccidentallyOnPurpose
 <4935931+AccidentallyOnPurpose@users.noreply.github.com4935931+AccidentallyOnPurpose@users.noreply.github.com>
Date: Tue, 24 Dec 2019 20:51:35 +0800
Subject: [PATCH] fix

---
 gpt2generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpt2generator.py b/gpt2generator.py
index a1c3e72..bae72ad 100644
--- a/gpt2generator.py
+++ b/gpt2generator.py
@@ -116,7 +116,7 @@ class GPT2Generator:
         self.model_dir = "models"
         self.checkpoint_path = os.path.join(self.model_dir, self.model_name)
         assert os.path.exists(self.checkpoint_path), "Make sure to download the pytorch v5 model and put it in " + self.checkpoint_path
-        self.checkpoint_path = "gpt2"
+        # self.checkpoint_path = "gpt2"
         self.device = torch.device("cuda" if not CPU else "cpu")
         logger.info("Using device={}, checkpoint={}, dtype={}".format(self.device, self.checkpoint_path, self.dtype))