Always copy metadata files

minimaxir · Aug 28, 2019 · e161bbe · e161bbe
1 parent f9734d2
commit e161bbe
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py
@@ -156,13 +156,12 @@ def maketree(path):
     maketree(checkpoint_path)
     files = [f for f in os.listdir(checkpoint_path)]
     for file in ['hparams.json', 'encoder.json', 'vocab.bpe']:
-        if file not in files:
-            try:
-                shutil.copyfile(os.path.join(model_dir, model_name, file),
-                                os.path.join(checkpoint_path, file))
-            except FileNotFoundError as fnf_error:
-                print("You need to download the GPT-2 model first via download_gpt2()")
-                raise(fnf_error)
+        try:
+            shutil.copyfile(os.path.join(model_dir, model_name, file),
+                            os.path.join(checkpoint_path, file))
+        except FileNotFoundError as fnf_error:
+            print("You need to download the GPT-2 model first via download_gpt2()")
+            raise(fnf_error)
 
     enc = encoder.get_encoder(checkpoint_path)
     hparams = model.default_hparams()