From d1f412323faba0399b1dfe1018f61226b7f8a239 Mon Sep 17 00:00:00 2001 From: tidedra Date: Thu, 8 Aug 2024 13:18:13 +0800 Subject: [PATCH] fix environment variable export bug for MultiNodeRunner --- deepspeed/launcher/multinode_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/launcher/multinode_runner.py b/deepspeed/launcher/multinode_runner.py index ce58deadc281..f8d1e00203fa 100644 --- a/deepspeed/launcher/multinode_runner.py +++ b/deepspeed/launcher/multinode_runner.py @@ -34,7 +34,7 @@ def get_cmd(self, environment, active_resources): """Return the command to execute on node""" def add_export(self, key, var): - self.exports[key.strip()] = var.strip() + self.exports[key.strip()] = f"\"{var.strip()}\"" def parse_user_args(self): return self.args.user_args