Skip to content

Commit d85a4b9

Browse files
committed
Update pytorch_runtime.py to fix environment variable bug.
1 parent b380572 commit d85a4b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ads/jobs/builders/runtimes/pytorch_runtime.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def run(self, dsc_job, **kwargs):
206206
envs = {}
207207
# Huggingface accelerate requires machine rank
208208
envs["RANK"] = str(i)
209-
envs["WORLD_SIZE"] = replicas
209+
envs["WORLD_SIZE"] = str(replicas)
210210
if main_run:
211211
envs["MAIN_JOB_RUN_OCID"] = main_run.id
212212
name = replica_kwargs.get("display_name")

0 commit comments

Comments
 (0)