update vocab file loc

extract.py

@@ -13,7 +13,7 @@ tarxz_path = "./openwebtext.tar.xz"
 folder_path = "./openwebtext"
 output_file_train = "./openwebtext/train_split.txt"
 output_file_val = "./openwebtext/val_split.txt"
-vocab_file = "./openwebtext/vocab.txt"
+vocab_file = "./vocab.txt"
 
 if not os.path.exists(tarxz_path):
     print("Please download the openwebtext.tar.xz file from:")

train_gpt_openwebtext.py

@@ -19,10 +19,10 @@ n_head = 8
 n_layer = 8
 dropout = 0.2
 
-if not os.path.exists("./openwebtext/vocab.txt") or not os.path.exists("./openwebtext/train_split.txt") or not os.path.exists("./openwebtext/val_split.txt"):
+if not os.path.exists("./vocab.txt") or not os.path.exists("./openwebtext/train_split.txt") or not os.path.exists("./openwebtext/val_split.txt"):
     raise Exception("Please run extract.py first")
 chars = ""
-with open("./openwebtext/vocab.txt", 'r', encoding='utf-8') as f:
+with open("./vocab.txt", 'r', encoding='utf-8') as f:
     text = f.read()
     chars = sorted(list(set(text)))