[bug] fix model idx

OptimalScale · Jul 21, 2024 · 998e1cb · 998e1cb
1 parent 55753d5
commit 998e1cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lmflow/pipeline/iterative_dpo_aligner.py b/src/lmflow/pipeline/iterative_dpo_aligner.py
@@ -59,7 +59,7 @@ def align(
                 target_model_args = self.model_args
             else:
                 target_model_args = copy.deepcopy(self.model_args)
-                target_model_args.model_name_or_path = str(self.workspace_path/f"iteration_{iter_idx-1}"/"model")
+                target_model_args.model_name_or_path = str(self.workspace_path/f"iteration_{iter_idx}"/"model")
 
             self._align_single_iteration(
                 iteration_name=f"iteration_{iter_idx+1}",