fix switch of --account + --comment

EleutherAI · Jan 19, 2024 · d10a651 · d10a651
1 parent 86bc680
commit d10a651
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepspeed/launcher/multinode_runner.py b/deepspeed/launcher/multinode_runner.py
@@ -424,10 +424,10 @@ def get_cmd(self, environment, active_resources):
         ] + split(self.args.launcher_args)
 
         if getattr(self.args, 'comment', ''):
-            srun_cmd += ['--account', self.args.comment]
+            srun_cmd += ['--comment', self.args.comment]
 
         if getattr(self.args, 'account', ''):
-            srun_cmd += ['--comment', self.args.comment]
+            srun_cmd += ['--account', self.args.account]
 
         if self.args.include != "":
             srun_cmd.append('--nodelist')