Upload train_memory_agent.py with huggingface_hub

train_memory_agent.py

@@ -16,10 +16,9 @@ from peft import LoraConfig
 from trl import SFTTrainer, SFTConfig
 
 print("Loading dataset...")
-dataset = load_dataset("erik1988/elias-memory-agent-sft-v1", split="train")
+dataset = load_dataset("erik1988/way2agi-memory-agent-sft", data_files="memory-agent-sft-v3-merged.jsonl", split="train")
 print(f"Dataset loaded: {len(dataset)} examples")
 print(f"Columns: {dataset.column_names}")
-print(f"First example keys: {dataset[0].keys()}")
 
 dataset_split = dataset.train_test_split(test_size=0.1, seed=42)
 train_dataset = dataset_split["train"]
@@ -55,7 +54,7 @@ config = SFTConfig(
     gradient_checkpointing=True,
     report_to="trackio",
     project="way2agi-memory-agent",
-    run_name="memory-agent-sft-v1-qwen1.5b",
+    run_name="memory-agent-sft-v3-qwen1.5b",
 )
 
 print("Initializing trainer...")