Update pytorch_runtime.py to set RANK and WORLD_SIZE env vars.

Author: qiuosier

ads/jobs/builders/runtimes/pytorch_runtime.py

@@ -205,7 +205,8 @@ def run(self, dsc_job, **kwargs):
                 if not envs:
                     envs = {}
                 # Huggingface accelerate requires machine rank
-                envs["OCI__NODE_RANK"] = str(i)
+                envs["RANK"] = str(i)
+                envs["WORLD_SIZE"] = replicas
                 if main_run:
                     envs["MAIN_JOB_RUN_OCID"] = main_run.id
                 name = replica_kwargs.get("display_name")