accomodate renaming of include_tokens_per_second to include_num_input_tokens_seen

Author: yash4242

tests/build/test_launch_script.py

@@ -51,7 +51,8 @@
     "warmup_ratio": 0.03,
     "lr_scheduler_type": "cosine",
     "logging_steps": 1,
-    "include_tokens_per_second": True,
+    # "include_tokens_per_second": True,
+    "include_num_input_tokens_seen": True,
     "packing": False,
     "response_template": "\n### Label:",
     "dataset_text_field": "output",

tests/test_sft_trainer.py

@@ -124,7 +124,8 @@
     warmup_ratio=0.03,
     lr_scheduler_type="cosine",
     logging_steps=1,
-    include_tokens_per_second=True,
+    # include_tokens_per_second=True,
+    include_num_input_tokens_seen=True,
     packing=False,
     max_seq_length=4096,
     save_strategy="epoch",
@@ -140,7 +141,8 @@
     warmup_ratio=0.03,
     lr_scheduler_type="cosine",
     logging_steps=1,
-    include_tokens_per_second=True,
+    # include_tokens_per_second=True,
+    include_num_input_tokens_seen=True,
     packing=False,
     max_seq_length=4096,
     save_strategy="epoch",