Update train.py

train.py

@@ -2,8 +2,11 @@ import os, pickle, json, torch
 import torch.nn as nn
 from model import GPT, GPTConfig
 
-with open("data/ai_gf/input.txt", "r", encoding="utf-8") as f:
-    text = f.read()
+# Load both original and extra data
+with open("data/ai_gf/input.txt", "r", encoding="utf-8") as f1, \
+     open("data/ai_gf/input_extra.txt", "r", encoding="utf-8") as f2:
+    text = f1.read() + "\n\n" + f2.read()
+
 
 chars = sorted(list(set(text)))
 vocab_size = len(chars)