Training script with WorldModel/Memory integration

train.py

@@ -20,6 +20,16 @@ from peft import set_peft_model_state_dict
 # CRITICAL: Import unsloth FIRST before any other ML libraries
 from unsloth import FastLanguageModel
 
+# Import WorldModel and Memory (for full implementation)
+try:
+    from world_model import WorldModelSimulator
+    from memory import MemoryLedger
+
+    WORLD_MODEL_AVAILABLE = True
+except ImportError:
+    WORLD_MODEL_AVAILABLE = False
+    print("Note: WorldModel/Memory not available in this environment")
+
 # ─── Auth & Config ────────────────────────────────────────────────────────────
 HF_TOKEN = os.environ["HF_TOKEN"]
 HF_USERNAME = os.environ.get("HF_USERNAME", "YashashMathur")