[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

applications/ColossalChat/coati/dataset/tokenization_utils.py

@@ -392,4 +392,4 @@ def tokenize_kto(
         "label": data_point["label"],
         "input_id_decode": decoded_full_prompt,
         "completion_decode": decoded_completion,
-    }
+    }

applications/ColossalChat/coati/trainer/dpo.py

@@ -356,4 +356,4 @@ class DPOTrainer(SLTrainer):
         os.makedirs(self.save_dir, exist_ok=True)
         with open(os.path.join(self.save_dir, f"eval_result_epoch{epoch}.txt"), "w") as f:
             f.write(msg)
-        step_bar.close()
+        step_bar.close()

applications/ColossalChat/coati/trainer/kto.py

@@ -346,4 +346,4 @@ class KTOTrainer(SLTrainer):
         os.makedirs(self.save_dir, exist_ok=True)
         with open(os.path.join(self.save_dir, f"eval_result_epoch{epoch}.txt"), "w") as f:
             f.write(msg)
-        step_bar.close()
+        step_bar.close()

applications/ColossalChat/coati/trainer/orpo.py

@@ -323,4 +323,4 @@ class ORPOTrainer(SLTrainer):
         os.makedirs(self.save_dir, exist_ok=True)
         with open(os.path.join(self.save_dir, f"eval_result_epoch{epoch}.txt"), "w") as f:
             f.write(msg)
-        step_bar.close()
+        step_bar.close()

applications/ColossalChat/examples/README.md

@@ -903,4 +903,4 @@ For details, see [`inference/`](https://github.com/hpcaitech/ColossalAI/tree/mai
 ## Attention
 
 
-The examples are demos for the whole training process. You need to change the hyper-parameters to reach great performance.
+The examples are demos for the whole training process. You need to change the hyper-parameters to reach great performance.

applications/ColossalChat/examples/training_scripts/train_kto.py

@@ -375,4 +375,4 @@ if __name__ == "__main__":
         os.makedirs(os.path.dirname(args.config_file), exist_ok=True)
         with open(args.config_file, "w") as f:
             json.dump(args.__dict__, f, indent=4)
-    train(args)
+    train(args)

applications/ColossalChat/examples/training_scripts/train_orpo.py

@@ -340,4 +340,4 @@ if __name__ == "__main__":
         os.makedirs(os.path.dirname(args.config_file), exist_ok=True)
         with open(args.config_file, "w") as f:
             json.dump(args.__dict__, f, indent=4)
-    train(args)
+    train(args)

applications/ColossalChat/requirements.txt

@@ -20,4 +20,4 @@ datasets
 ninja==1.11.1
 sentencepiece==0.1.99
 flash-attn
-tiktoken
+tiktoken

applications/ColossalChat/tests/test_train.sh

@@ -640,4 +640,4 @@ for lora_rank in ${LORA_RANK[@]}; do
             fi
         done
     done
-done
+done