fixed code formatting

microsoft · Oct 14, 2024 · 82c59ab · 82c59ab
1 parent 144e913
commit 82c59ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/applications/DeepSpeed-Chat/dschat/rlhf/rlhf_engine.py b/applications/DeepSpeed-Chat/dschat/rlhf/rlhf_engine.py
@@ -269,8 +269,8 @@ def _init_reward(self, critic_model_name_or_path):
             zero_stage = 0
 
         ds_config = get_eval_ds_config(offload=self.args.offload_reward_model,
-                                            dtype=self.args.dtype,
-                                            stage=zero_stage)
+                                       dtype=self.args.dtype,
+                                       stage=zero_stage)
 
         # We need to set train batch size and micro batch size here to pass the sanity check of DeepSpeed engine.
         ds_config[