manually set roles to train on

2026-01-08 21:28:05 -05:00 · 2025-12-21 22:09:18 -05:00
parent cf01fd29ae
commit 1811a907f7
4 changed files with 29 additions and 12 deletions
--- a/train/configs/functiongemma-270m.yml
+++ b/train/configs/functiongemma-270m.yml
@@ -308,7 +308,7 @@ datasets:
 - path: /workspace/data/datasets/sample.jsonl
  ds_type: json
  type: chat_template
-  # roles_to_train: [ "assistant" ]
+  roles_to_train: []
  message_field_training: train_on_turn

 dataset_prepared_path: /workspace/data/datasets/prepared/