Fix: max_length -> max_seq_length for trl 0.15.2 (verified all configs locally)

train_on_hf.py

@@ -121,7 +121,7 @@ def train_world_model(data_dir: Path, max_samples: int = 10000):
         save_steps=500,
         save_total_limit=2,
         fp16=True,
-        max_length=512,
+        max_seq_length=512,
         report_to="none",
         disable_tqdm=False,
         dataloader_num_workers=0,
@@ -229,7 +229,7 @@ def train_strategist(data_dir: Path, max_samples: int = 10000):
         gradient_accumulation_steps=2,
         learning_rate=2e-4,
         fp16=True,
-        max_length=512,
+        max_seq_length=512,
         logging_steps=5,
         save_steps=100,
         report_to="none",